Tim M b22c929a64 Corrected additional `.md` files with new image		2 tahun lalu
..
media	5f54a1618e YOLOv6 Custom Object Detection Training	2 tahun lalu
notebooks	5f54a1618e YOLOv6 Custom Object Detection Training	2 tahun lalu
README.md	b22c929a64 Corrected additional `.md` files with new image	2 tahun lalu

YOLOv6 Custom Dataset Training – Underwater Trash Detection

This repository contains the code for the LearnOpenCV blog post YOLOv6 Custom Dataset Training - Underwater Trash Detection

We train several YOLOv6 models on real-life and challenging dataset to detect underwater trash on deep ocean floors.

The notebooks directory contains the Jupyter Notebook which will run end-to-end with one click. If you have Ubuntu, you can run it locally. Else, you may also upload it to Colab and run it. The notebook will train three different models for different experiments.
The three training experiments are:
- Fine tuning YOLOv6 Nano model.
- Fine tuning YOLOv6 Small model.
- Fine tuning YOLOv6 Large model.

Download the trained weights from here.

AI Courses by OpenCV

Want to become an expert in AI? AI Courses by OpenCV is a great place to start.