Skip to content

Latest commit

 

History

History
6 lines (5 loc) · 267 Bytes

File metadata and controls

6 lines (5 loc) · 267 Bytes

cone_object_detection_bounding_boxes

Identifying cones in a video, identifying which color the cone is, and drawing a bounding box around the cone

Used PyTorch to train YOLOv5 nano model Trained on 10 custom training images/annotations 300 epochs, batch size = 10