/Object-detection-yolo-v5

Primary LanguageJupyter NotebookGNU General Public License v3.0GPL-3.0

Object Detection yolo v5

This is a demo for the Yolo v5 with a custom dataset that is produced by Open Images Dataset v4. The code will be found in the colab and it is practical part of the paper of Object Detection overview.

This colab notebook contains the creation of the custom dataset, the conversion from the OID labels to the yolov5 compatible and the training of the model with the different models.