javier-marti-isasi / yolov5m-training-using-finetuning-and-data-augmentation-on-a-custom-labeled-dataset Goto Github PK
View Code? Open in Web Editor NEWObject detection YOLOv5m finetuning on custom dataset with PyTorch. The custom dataset was labeled in CVAT and preprocessed in Roboflow. Data augmentation is implemented in the last experiments. Inference on images.