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

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，膜拜

这么快就做出来了，tql！

猛呀老哥，tql

tql,tql,tql

tql，膜拜

你用什么解压的数据集呀。解压不了呀

qxy 发表于 2019-12-20 11:06
你用什么解压的数据集呀。解压不了呀

用的winrar解压

tql 膜拜

qxy 发表于 2019-12-20 11:06
你用什么解压的数据集呀。解压不了呀

7zip比较稳