赛事链接: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写自己的配置文件
<