roboflow / notebooks

Examples and tutorials on using SOTA computer vision models and techniques. Learn everything from old-school ResNet, through YOLO and object-detection transformers like DETR, to the latest models like Grounding DINO and SAM.
https://roboflow.com/models
5.6k stars 882 forks source link

Clean up: Yolov5 Object Detection #329

Closed LinasKo closed 1 month ago

LinasKo commented 1 month ago

Description

Updated Yolov5, making sure the model can be trained and deployed to Roboflow.

The major changes is the removal of custom config, where model head layers were defined explicitly, rather than using a predefined confugration such as yolov5s.

Type of change

Please delete options that are not relevant.

How has this change been tested, please provide a testcase or example of how you tested the change?

I ran the entire notebook, training and deploying the model to Roboflow.

Any specific deployment considerations

For example, documentation changes, usability, usage/costs, secrets, etc.

Docs