Skip to content

Latest commit

 

History

History
27 lines (20 loc) · 1.02 KB

File metadata and controls

27 lines (20 loc) · 1.02 KB

VisDrone_YOLOv5

Building Custom Object Detection model on VisDrone2021 dataset using state of the art YoloV5

Files:

1.convert.py: normalising cooridnates for bounding boxes
2.best.pt: best weight for our trained model
2.VisDrone_YOLOv5.ipynb: Training custom object detector for 11 classes that include:

{ pedestrian(1),people(2), bicycle(3), car(4), van(5), truck(6), tricycle(7), awning-tricycle(8), bus(9), motor(10),others(11) }

Results

alt text

Graphs

alt text

Train_batch images

alt text

Test Images

alt text

alt text