labelme使用labelme2voc.py忽略部分标签生成VOC/coco类型数据集(Ignore a label in json file)
创始人
2024-05-13 06:15:59
0

一:背景和问题描述

这是一个比较小众,且查找全网未有答案的问题。

在使用labelme标注数据之后,生成了json文件。在利用json文件生成可用于训练分割模型的voc或者coco数据集时,可以使用labelme中自带的示例文件labelme2voc.py,如下

但是我在制作数据集时为了数据集更加多样可用,标注了很多种暂时用不到的标签,转换时忽视指定的标签,如何做呢。

GitHub中有人提出了这样的问题,但是没有详细的解决办法,如下,

问题详情:

github见https://github.com/wkentaro/labelme/discussions/1097

Hi, I did what you say but got an error
Traceback (most recent call last):
File "labelme2voc.py", line 139, in
main()
File "labelme2voc.py", line 102, in main
cls, ins = labelme.utils.shapes_to_label(
File "/Users/azzz/.pyenv/versions/3.8.3/lib/python3.8/site-packages/labelme/utils/shape.py", line 69, in shapes_to_label
cls_id = label_name_to_value[cls_name]
KeyError: 'truck'
truck is the label I want to ignore
——————————————————————————————————
Hi, I have the same error, too.
Traceback (most recent call last):
File "./labelme2voc.py", line 106, in
main()
File "./labelme2voc.py", line 88, in main
label_name_to_value=class_name_to_id,
File "/Users/tajimakeijiro/anaconda3/envs/labelme/lib/python3.6/site-packages/labelme/utils/shape.py", line 68, in shapes_to_label
cls_id = label_name_to_value[cls_name]
KeyError: 'wall'
I wanted to ignore the class 'wall', so I removed it from labels.txt.
But I got this error.

二、解决办法

查看了labelme2voc.py源码,因此我们直接在源码中进行修改,修改方式比较愚笨且直接,但是能够解决问题。

假设我们所有的标签'crack,shedding,damage,face1,face2,face3,其中face1,face2,face3,是我们不需要的。 正常情况下我们转换所有标签,修改labels.txt,即

__ignore__
_background_
crack
shedding
damage
face1
face2
face3
background

如果我们不需要face1,face2,face3,从上面中删除face1,face2,face3会报错。我们只需要将如下代码添到labelme2voc.py中的class_names = tuple(class_names) 下面,即可达到忽略face1,face2,face3的目的。此时修改labels.txt操作变得可有可无了。

class_names=['_background_','crack','shedding','damage','face1','face2','face3','background']
class_name_to_id={'__ignore__':-1,'face1':0,'face2':0,'face3':0,'_background_':0,'crack':1,'shedding':2,'damage':3,'background':4}

如图所示:

三:原理和解释

在转换过程中,背景(background)用值0表示,因此我们如果忽视标签'face1','face2','face3',我们将其也设置为0即可达到目的。

class_name表示所有的类别。其中_background_必须要有,

而'crack,shedding,damage,face1,face2,face3,background是类别。

class_name_to_id表示这些类别的数字表示。其中的'__ignore__':-1和'_background_':0必须要有;

而待忽略的标签,将其值设置为背景的值,即0即可,即'face1':0,'face2':0,'face3':0;,不忽略的标签,后面依次按照想要的顺序写出来并赋值即可,从1开始。

四:总结

在使用labelme2voc.py时,进行调试,可以更好地理解添加上面两行代码的原理,这也是我使用该方法的思路来源。coco类型应该类似,可以自行调试或者构造出代码。

相关内容

热门资讯

教师节朗诵晚会串词 教师节朗诵晚会串词  主持词需要富有情感,充满热情,才能有效地吸引到观众。在当今不断发展的世界,越来...
趣味运动会主持稿 趣味运动会主持稿(通用7篇)  在充满活力,日益开放的今天,很多地方都会使用到主持稿,主持稿是主持人...
鼠年文艺晚会的主持词 鼠年文艺晚会的主持词  1、喜鹊枝头叫,猴年已来到。好运来报道,健康身边绕。跨羊威武耀,进财装元宝。...
订婚仪式主持词 订婚仪式主持词(通用16篇)  主持词的写作需要将主题贯穿于所有节目之中。在人们积极参与各种活动的今...
文化艺术节闭幕词 文化艺术节闭幕词(精选6篇)  在日新月异的现代社会中,我们都可能会用到闭幕词,闭幕词的作用是辅助讲...
幼儿园签约仪式主持词 幼儿园签约仪式主持词  主持词没有固定的格式,他的最大特点就是富有个性。现今社会在不断向前发展,主持...
电影后来的我们台词 电影后来的我们台词  电影《后来的我们》可以说是风波不断,那么后来的我们经典语录台词有哪些?下面是小...
音乐串词 音乐串词甲:金牛报春来! 又是瑞雪飘飞的季节了,我们打一小学这个大家庭每到这时候总会欢聚一堂。 共庆...
不差钱的经典台词 不差钱的经典台词  赵本山:苏格兰调情。  小沈阳:人家是纯爷们。  丫蛋:洪湖水浪打浪,长江后浪推...
《终结者2》的经典台词 《终结者2》的经典台词  1.I’ll be back。  我会回来的。  2.I need you...
婚宴新娘致辞 婚宴新娘致辞(合集15篇)  无论在学习、工作或是生活中,大家都对致辞很是熟悉吧,在各种重大的庆典、...
单位领导证婚词 单位领导证婚词尊敬的各位来宾、各位亲朋好友,女士们、先生们:大家中--午--好!今天,艳阳高照,天赐...
诵经典唱红歌主持词 诵经典唱红歌主持词  女:各位领导、各位来宾,  男:老师们、同学们,  合:大家好!  女:五月良...
笑傲江湖之东方不败台词 笑傲江湖之东方不败台词大全  天下风云出我辈,  一入江湖岁月催。  皇图霸业谈笑中,  不胜人生一...
学校元宵联欢晚会的主持词 学校元宵联欢晚会的主持词(精选6篇)  利用在中国拥有几千年文化的诗词能够有效提高主持词的感染力。我...
新婚婚礼主持词 新婚婚礼主持词  主持词的写作要突出活动的主旨并贯穿始终。我们眼下的社会,各种集会的节目都通过主持人...
酒店开业致辞 酒店开业致辞(精选20篇)  在日常学习、工作抑或是生活中,大家对致辞都再熟悉不过了吧,致辞具有有张...
主持人台词 主持人台词大全  导语:在剧作中,台词是一种重要的手段和方法,可以用来刻画人物的性格特征,可以用来展...
封顶仪式致辞 封顶仪式致辞  在平凡的学习、工作、生活中,大家都写过致辞吧,致辞具有很强的实用性和针对性。究竟什么...
唯美中式婚礼主持词 唯美中式婚礼主持词五篇  活动对象的不同,主持词的写作风格也会大不一样。我们眼下的社会,主持词是活动...