
YOLOV5 肋骨骨折检测改进项目【采用ResNet作为骨干网络】(含数据、代码及预训练模型)
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目基于YOLOv5框架开发了肋骨骨折检测系统,并采用了ResNet作为其核心骨干网络,旨在提升检测精度。该项目提供完整数据集、源代码以及预训练模型,便于科研人员进行二次开发和应用研究。
YOLOV5 改进实战项目【更换骨干网络为resnet】用于肋骨骨折检测,包含代码、数据集、训练好的权重参数。经测试,可以直接使用该代码。
项目总大小:686MB
本项目将yolov5的骨干网络替换成了官方实现的resnet,并进行了30个epoch的简单训练,map指标为0.37,而map0.5:0.95则达到了0.19。这里仅使用了30个epoch用于测试,因此模型还未完全收敛;增加更多的轮次可以提高网络性能。
【如何训练】与yolov5相同的方法进行训练:先准备好数据集,并修改yaml文件中的类别信息后即可开始训练。
【数据集介绍】
肋骨骨折图像数据共包含五种类别:移位的肋骨骨折、非移位的肋骨骨折、扣肋骨折、节段性肋骨骨折和不确定类型肋骨骨折。
- 训练集包括4618张图片及其对应的标签txt文件;
- 验证集则由1076张图片及相应的标签txt文件组成。
全部评论 (0)
还没有任何评论哟~


