← all repositories

DataXujing/YOLO-v5

A PyTorch YOLOv5 tutorial for training object detection models on custom datasets in medical endoscopy.

971 stars Jupyter Notebook Computer VisionML Frameworks
YOLO-v5
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.