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

[复制链接]
梯队队员溜溜球です
2019-12-18 23:37:21 显示全部楼层

马上注册,玩转Robomaster!

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
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

视频截图
捕获.PNG

2600次迭代 loss 1.88
chart.png


跳转到指定楼层

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

[复制链接]
梯队队员541594891
2020-1-11 14:54:43 显示全部楼层
溜溜球です 发表于 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>
我用了自己做的标注文件,名字是一样的他就可以处理出来
你知道这是什么问题吗

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

[复制链接]
梯队队员溜溜球です
 楼主| 2020-1-11 14:03:00 显示全部楼层
541594891 发表于 2020-1-9 17:21
是用的yolo吗,官方给的那些xml里的armor没有difficult,我在做yolo的时候没法用darknet给的脚本作预处理, ...

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

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

[复制链接]
梯队队员溜溜球です
 楼主| 2020-1-12 14:49:22 显示全部楼层
541594891 发表于 2020-1-11 14:54
我在运行那个py的时候他提示我
Traceback (most recent call last):
  File "voc_label.py", line 54, in ...

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

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

[复制链接]
工程lazyzabre
2019-12-18 23:46:08 显示全部楼层
大佬tql,膜拜

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

[复制链接]
正式队员HuskyRye
2019-12-19 04:51:01 显示全部楼层
这么快就做出来了,tql!

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

[复制链接]
顾问1145284121
2019-12-19 04:57:52 显示全部楼层
猛呀老哥,tql

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

[复制链接]
英雄陶驹驹
2019-12-19 09:20:32 显示全部楼层
tql,tql,tql
回复

使用道具 举报

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

[复制链接]
工程Edwin 蒙
2019-12-19 13:03:01 显示全部楼层
tql,膜拜
回复

使用道具 举报

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

[复制链接]
梯队队员qxy
2019-12-20 11:06:29 显示全部楼层
你用什么解压的数据集呀。解压不了呀

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

[复制链接]
梯队队员minamikotori
2019-12-20 13:46:05 显示全部楼层
qxy 发表于 2019-12-20 11:06
你用什么解压的数据集呀。解压不了呀

用的winrar解压

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

[复制链接]
顾问西瓜柠檬棒棒糖
2019-12-20 20:00:07 显示全部楼层
tql 膜拜
回复

使用道具 举报

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

[复制链接]
正式队员苏锌雨
2019-12-21 13:01:05 显示全部楼层
qxy 发表于 2019-12-20 11:06
你用什么解压的数据集呀。解压不了呀

7zip比较稳
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

触屏版 | 电脑版

Copyright © 2024 RoboMasters 版权所有 备案号 粤ICP备2022092332号

快速回复 返回顶部 返回列表