
深度学习与目标检测:利用自备数据集、YOLOX及模型剪枝实现轻量化训练
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本研究探讨了基于自备数据集和YOLOX框架进行目标检测的方法,并通过模型剪枝技术优化,实现了模型的轻量化训练。
采用PyTorch深度学习环境,目标检测框架为YOLOX,并支持模型剪枝功能。主要功能包括:
1. 支持训练自定义数据集。
2. 具备图像推理能力。
3. 可进行视频推理操作。
4. 提供mAP测试功能。
5. 支持对任意层的剪枝操作。
6. 适用于多种模型架构的应用场景。
7. 包含微调训练支持,并附带readme文件,便于快速上手。
代码成熟且经过多人验证,适合研究人员、研究生和大学生等深度学习研究者使用。
全部评论 (0)
还没有任何评论哟~


