YOLOv3训练自己的数据

关键点在于:弄清YOLOv3数据集的组织形式,需要按照这个生成数据。

一. 训练环境
1. YOLOv3编译:OpenCV+GPU
git clone https://github.com/pjreddie/darknet
cd darknet
vim ...

2. 下载预训练权重(只包含卷积层)
    wget https://pjreddie.com/media/files/darknet53.conv.74 

二. 数据准备
0. 因为YOLOv3提供了一个将VOC数据集转成自己训练的数据集格式的工具:scripts/voc_label.py,
因此,训练数据只需要按VOC数据集的格式准备就好, 再执行voc_label.py即可
另一种思路,直接生成YOLO格式的?YOLO的格式更简洁些。
1. VOC2012
http://host.robots.ox.ac.uk/pascal/VOC/voc2012/
研究下文件组织结构
Annotations:用于存放与图片对应的XML文件
JPEGImages: 存放所有的图片,png是否可以?
2. 对自己的图片数据打标,按照YOLOv3的方式准备数据集
人工选取: l78z/projects/video_roi/video_roi_manual.py
自动保存: l78z/projects/video_roi/video_roi_crop.py
其他:labelImg: https://github.com/tzutalin/labelImg#macos
很多教程是基于次标注工具产出数据,再转成yolo训练数据
该工具支持直接生成YOLO格式,
基于cvui搞个工具?
ctrl+t , 打开新视频
左键选择,右键取消
3. 使用 scripts/voc_label.py 生成

三、 修改YOLOv3的配置文件
1. data/voc.names #修改成自己的分类
2. cfg/voc.data
classes = N #(N为自己的分类数量,如有10类不同的对象,N = 10)
train = /home/XXX/darknet/trainImagePath.txt # 训练集完整路径列表
valid = /home/XXX/darknet/validateImagePath.txt # 测试集完整路径列表
names = data/voc.names # 类别文件
backup = backup #(训练结果保存在darknet/backup/目录下)
3. cfg/yolov3-voc.cfg
classes = N (N为自己的分类数)
filters = 3*(classes+1+4), 修改每一个[yolo]层(一共有3处)之前的filters.
(可选) 修改训练的最大迭代次数, max_batches = N

四、 YOLOv3训练&测试
训练:
./darknet detector train cfg/voc.data cfg/yolov3-voc.cfg darknet53.conv.74
测试:
./darknet detector test cfg/voc.data cfg/yolov3-voc.cfg backup/yolov3-voc_final.weights 01.jpg

五、题外话,说说视频标注工具的设计
1. 读取video_list.txt,创建目录文件,下载视频,存放至约定的文件夹,文件价格结构:
video_list.txt, video_id \t video_download_url
Videos/ 用于存放下载的视频
Manual/ 用于存放用户选择的图片、区域、区域裁剪图
JPEGImages/ 存放自动生成的图片
Annotations/ 用于图片对应的选取信息
class.txt 所有分类汇总
class/ 选取区域的图像 ?
class1/file1.jpg
2. 自动定位至还没有标注数据的,显示标注进度。
3. 快捷键设定:按游戏的设定w前进帧,s后退帧,d下一个,a上一个(上下切换视频,左右调节帧,方向键位于右侧,影响鼠标操作),上下切换时,自动保存。
4. 拖拽选中区域,要有x/y辅助线,右键取消上一次选中,
5. 选中结束,弹窗,类名,设置类名时,除了文字外有对应的缩略图
6. 保存VOC格式, 全部结束后,自动生成YOLO格式。

参考文档:
https://blog.csdn.net/dcrmg/article/details/81296520
https://www.jianshu.com/p/f4518fe04da1
https://www.jianshu.com/p/91eafe0f3719
https://www.cnblogs.com/pprp/p/9525508.html

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,125评论 6 498
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,293评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 162,054评论 0 351
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,077评论 1 291
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,096评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,062评论 1 295
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,988评论 3 417
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,817评论 0 273
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,266评论 1 310
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,486评论 2 331
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,646评论 1 347
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,375评论 5 342
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,974评论 3 325
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,621评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,796评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,642评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,538评论 2 352

推荐阅读更多精彩内容

  • 文章于2020年已更新 https://www.jianshu.com/p/2206db894b28 工具准备 D...
    绍重先阅读 21,895评论 0 9
  • refer Yolo darknet训练自己的数据集教程(Newest 2016.12.23) 经过两天的折腾终于...
    angerskon阅读 2,565评论 0 3
  • 不知不觉时钟来到了深夜一点。先前深夜里的自己多半有些寂寥、有些许孤独,静静地听着夜的声响五味杂陈。可人生年轮一圈一...
    一直奋斗的虾阅读 277评论 0 0
  • 深夜加班回家,电梯到了,走进去,叮咚!居然超重,“真是的,这破电梯,又乱显示。”抱怨起来,下到下一楼,有人按停,电...
    许你故作姿态阅读 83评论 0 1
  • 我的委内瑞拉朋友来北京,从来没有经历过这么寒冷的天气。虽然冷,我却感受到了南美人的浪漫和热情,北京的故宫和长城见证...
    儒教教主牟阅读 631评论 0 11