- Arrange the dataset in the following structure.
carDamageDataset
|
|_ train
| |
| |_ labels
| | |_ .txt train images labels
| |
| |_ images
| |_ .jpg/.JPEG test images
|
|_ test / val (same as train)
- Clone Yolov5 GitHub repository, install dependencies and check PyTorch and GPU.
- Create the data.yaml for the dataset location and labels.
- Edit the global variables and run the notebook.