水下目标检测总结

本文介绍了水下目标检测的赛事及数据集标注,选用YOLOv5模型进行训练,并分享了训练配置、模型选择及训练过程。在泰坦显卡上,YOLOv5s模型表现出速度快但小目标检测准确性不高的特点。通过颜色矫正和数据增强等方法,有望提升检测效果。最后,作者提到模型的轻量化特性适合嵌入式部署。
摘要由CSDN通过智能技术生成

赛事链接:https://www.flyai.com/d/UnderwaterDetection

数据集标注 voc格式:

数据处理参考:https://blog.csdn.net/Clay_Zhang/article/details/107079762

<?xml version="1.0" ?>
<annotation>
	<frame>000005</frame>
	<object>
		<name>holothurian</name>
		<bndbox>
			<xmin>865</xmin>
			<ymin>87</ymin>
			<xmax>1029</xmax>
			<ymax>182</ymax>
		</bndbox>
	</object>
	<object>
		<name>echinus</name>
		<bndbox>
			<xmin>1130</xmin>
			<ymin>286</ymin>
			<xmax>1260</xmax>
			<ymax>399</ymax>
		</bndbox>
	</object>

模型选择:

YOLOV5:https://github.com/ultralytics/yolov5
注意:pytorch版本必须是torch1.6否则模型跑不起来,yolov5使用了1.6版本才有的新的激活函数。
ps: pytorch最新的稳定版本1.7貌似今天发布了,支持cuda11 *

配置文件:

在data目录下写仿照coco.yaml写自己的配置文件

<
评论 16
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值