通过上一篇文章的分析,我们已经基本了解了VOC lmdb文件的格式,现在可以尝试将不同的标注格式,转为VOC格式。

此处安利一个我自己写的小工具,支持yolo格式转化为voc格式,同时支持lmdb文件的可视化。此处需要注意的是voc格式的标注一定不能超过图片大小。否则在运行create_data.sh时会报错。

https://github.com/Wang-zipeng/Deep_learning_format_tool

有了VOC格式的标注,可以调用/data/VOC0712/create_data.sh用于生成lmdb。但create_data中部分内容需要修改:

data_root_dir="$HOME/dataset/MOT"

用于设置数据集的根目录。

dataset_name="MOT17"

用于设置数据集名称。用于在example中生成数据集的软链接,可能还有其他作用。

mapfile="$root_dir/data/$dataset_name/labelmap_voc.prototxt"

用于设置类型名称与类型编号映射的文件。根据VOC的修改即可,此处需要注意的一点是类型 0,“background”需要保留。

for subset in test trainval

此处不需要修改,但我们在制作test.txt与trainval.txt时需要注意三点内容:

1)文件必须命名为test与trainval

2)test与trainval中的路径与data_root_dir中的路径必须可以构成完整路径。

3)test_name_size文件必须制作。

Logo

开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!

更多推荐