In this project, I created an upto 100 epochs custom dataset based fine tuned YOLO model using yolo11s.pt and then tested it on the image and video, after that I exported the cusotm trained model to ONNX.
YOLO11 training on custom data.
Roboflow traffic signal detection
IMPORTANT!! Do download the dataset in the YOLO format
- Saved best.pt file.
- Saved best.onnx file.
- For model's visualization, go here and upload your
*.ptor*.onnxfile.