JackTheProgrammer/Fine-tuned-YOLO11

GitHub: JackTheProgrammer/Fine-tuned-YOLO11

一个基于 YOLO11 并在交通信号数据集上微调的项目,解决了图像与视频中交通信号检测与导出的问题。

Stars: 0 | Forks: 0

# YOLO11s.pt 精调 在本项目中,我基于 [`yolo11s.pt`](model/yolo11s.pt) 创建了一个最多 100 个 epoch 的自定义数据集,并进行了精细微调,然后在图像和视频上进行了测试,之后将自定义训练模型导出为 ONNX 格式。 ## 开始 [在自定义数据上进行 YOLO11 训练](https://www.youtube.com/watch?v=A1V8yYlGEkI&t=1006s)。 ## 数据集 [Roboflow 交通信号检测](https://universe.roboflow.com/signal-sense-traffic-signal-detection/traffic-signals-mcmpm) *重要!!* 请务必以 **YOLO** 格式下载数据集 ## 结果 ![alt 100 个 epoch 的训练结果](https://static.pigsec.cn/wp-content/uploads/repos/2026/04/cda387df2d173906.png) ![alt 混淆矩阵](https://static.pigsec.cn/wp-content/uploads/repos/2026/04/570c463d41173914.png) ![alt 边界框 PR 曲线](https://static.pigsec.cn/wp-content/uploads/repos/2026/04/fbafe10124173919.png) ### 基于图像的预测结果 ![测试结果](https://static.pigsec.cn/wp-content/uploads/repos/2026/04/0332bf6009173924.jpg) ### 基于视频的预测结果 ![Alt 视频](https://raw.githubusercontent.com/JackTheProgrammer/Fine-tuned-YOLO11/main/runs/detect/video_predict/TrafficlightssequencetheorytestUKpart1-ezgif.com-optimize.gif) ### 保存的模型 * [保存的 best.pt 文件](runs/detect/yolo11s_custom_dataset/weights/best.pt)。 * [保存的 best.onnx 文件](runs/detect/yolo11s_custom_dataset/weights/best.pt)。 * 如需模型可视化,请[此处](https://netron.app/)上传您的 `*.pt` 或 `*.onnx` 文件。
标签:100 epochs, AI模型, CNCF毕业项目, fine tuning, ONNX导出, ONNX格式, Roboflow, YOLO11, YOLO格式, 交通信号检测, 交通信号识别, 图像识别, 图像预测, 模型优化, 模型训练, 模型部署, 深度学习, 目标检测, 目标检测模型, 自定义数据集, 视频分析, 视频检测, 计算机视觉, 逆向工具