
Pruned-OpenVINO-YOLO: 在OpenVINO嵌入式设备上部署优化的YOLOv3/YOLOv4/YOLOv4-tiny模型
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
Pruned-OpenVINO-YOLO项目致力于在OpenVINO框架下,为嵌入式设备提供轻量级且高效的YOLOv3、YOLOv4及YOLOv4-tiny版本的优化部署方案。通过模型剪枝技术显著减小模型大小并加速推理过程,同时保持高精度和实时性能,适用于资源受限环境下的目标检测任务。
修剪后的OpenVINO-YOLO
先决条件:
首先安装mish-cuda:测试平台为WIN10 + RTX3090 + CUDA11.2。
如果无法在设备上安装,可以尝试其他方式。
开发日志:
- 2021年2月25日:支持yolov4-tiny和yolov4-tiny-3l。请使用或修剪yolov4-tiny和yolov4-tiny-3l模型,并通过darknet调整修剪后的模型,无需加载权重。
- 2021年3月13日:更新了Mish-cuda支持,增强了代码对YOLOv4的适应性(训练更快、内存占用更少)。
介绍:
当在OpenVINO上部署YOLOv3和YOLOv4时,完整版模型FPS较低;而微型版本虽然提高了速度但准确性较差且稳定性不足。通常,完整结构设计用于复杂场景中检测80个或更多类别的目标,在实际应用中往往只需处理几个类别,并非所有场景都那么复杂。
本教程将分享如何修剪YOLOv3和YOLOv4模型以适应这些特定需求。
全部评论 (0)
还没有任何评论哟~


