DataXujing/YOLO-v5
A PyTorch YOLOv5 tutorial for training object detection models on custom datasets in medical endoscopy.

Velocity · 7d
+0.4
★ / day
Trend
→steady
star history
This repository provides a detailed guide for training YOLOv5 on custom datasets using PyTorch, with a specific focus on medical endoscopy object detection applications. It covers environment setup, dataset preparation, configuration files, and the training process. The project also references TensorRT acceleration for model inference.