【分享帖】西南民族大学 神经网络训练大疆开源数据集效果

SMU AJI战队 视觉组
配置：3700x 2070s
环境：Ubuntu18.04 opencv3.4.6 cuda10.2
模型：darknet
迭代次数：2600
loss：1.88
数据集：大疆开源 https://bbs.robomaster.com/thread-9678-1-1.html
视频：2019RM总决赛
由于训练次数较少,仅做测试用

测试视频链接https://www.bilibili.com/video/av79774867

视频截图

登录/注册后可看帖子

溜溜球です 发表于 2020-1-11 14:03
我就是用官方脚本做的预处理，你不能做预处理可能是因为官方给的文件有些名字有空格，python的那个脚本不 ...

我在运行那个py的时候他提示我
Traceback (most recent call last):
  File "voc_label.py", line 54, in <module>
    convert_annotation(year, image_id)
  File "voc_label.py", line 35, in convert_annotation
    difficult = obj.find('difficulty').text
AttributeError: 'NoneType' object has no attribute 'text'
，说那个找difficulty的那一句话出问题了，我去官方给的xml里找，发现armor里是没有difficulty的
<object>
        <name>armor</name>
        <armor_class>1</armor_class>
        <armor_color>blue</armor_color>
        <generated>0</generated>
        <is_incorrect>0</is_incorrect>
        <bndbox>
            <xmin>1418.25</xmin>
            <ymin>469.026</ymin>
            <xmax>1444.87</xmax>
            <ymax>482.967</ymax>
        </bndbox>
我用了自己做的标注文件，名字是一样的他就可以处理出来
你知道这是什么问题吗

541594891 发表于 2020-1-9 17:21
是用的yolo吗，官方给的那些xml里的armor没有difficult，我在做yolo的时候没法用darknet给的脚本作预处理， ...

我就是用官方脚本做的预处理，你不能做预处理可能是因为官方给的文件有些名字有空格，python的那个脚本不能识别带空格的文件名。会把每个空格都识别成一个文件，导致不能预处理，要么就是把带空格的文件都不处理，要么就是修改那个.py的代码

541594891 发表于 2020-1-11 14:54
我在运行那个py的时候他提示我
Traceback (most recent call last):
  File "voc_label.py", line 54, in ...

你可以直接把difficulty那段注释掉，让他不去寻找

tql 膜拜

大佬，请问一下，按照教程用darknet训练出来，检测图片没有检测框，这个是怎么回事啊

测试速度有多少呢楼主

大佬太强了

Railgun 发表于 2020-1-17 12:08
也是YOLOv3

没降到4以下过？也有可能是配置文件参数有问题

Railgun 发表于 2020-1-17 12:08
我们分了17类，训练迭代110000次，但是loss始终在4左右，怎么说。。。。

也是YOLOv3

我们分了17类，训练迭代110000次，但是loss始终在4左右，怎么说。。。。