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

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.