- A little bit about the dataset
- The result of the trained model - YOLO8 Medium custom
- The result of the trained model -
Fine-tuning
YOLO8 Medium custom - Conclusion
-
This dataset was created by extracting
frames
from12 short videos
. To enrich this data collection with a variety of images, such as blurry images, small-sized images, and others, it was decided to use the unmarked data set available on Kaggle. -
After the required number of frames was obtained, all images were annotated using the
Roboflow platform
.
As you can see, a trained model using fine tuning shows better results. Does not make mistakes when detecting an object, as it happens with a model without fine tuning