iqbalfarz / object-detection-custom-yolov7-tiny

Training YOLOv7-tiny on custom dataset

object-detection-custom-yolov7-tiny

Training YOLOv7-tiny on custom dataset

Problem Definition

Training YOLOv7-pretrained model on custom dataset for Person and Car Object.

Steps

• I created the Person and Car dataset from COCO and changed the data format to YOLOv7.
• How I created dataset? Please click to see the jupyter notebook
• Then, I created(uploaded) the dataset to Kaggle using Kaggle API and then
• Trained YOLOv7-tiny model for 60 epochs on custom dataset and did the inferencing on test images.
• Link to Kaggle Notebook: Kaggle Notebook!

Metrics

precision

Recall

F1-Curve

Confusion Matrix

Report

• See the report here: Click to see the report of training YOLOv7-tiny on custom dataset containing Person and Car