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类型应该类似,可以自行调试或者构造出代码。

相关内容

热门资讯

常用商务英语口语   商务英语是以适应职场生活的语言要求为目的,内容涉及到商务活动的方方面面。下面是小编收集的常用商务...
六年级上册英语第一单元练习题   一、根据要求写单词。  1.dry(反义词)__________________  2.writ...
复活节英文怎么说 复活节英文怎么说?复活节的英语翻译是什么?复活节:Easter;"Easter,anniversar...
2008年北京奥运会主题曲 2008年北京奥运会(第29届夏季奥林匹克运动会),2008年8月8日到2008年8月24日在中华人...
英语道歉信 英语道歉信15篇  在日常生活中,道歉信的使用频率越来越高,通过道歉信,我们可以更好地解释事情发生的...
六年级英语专题训练(连词成句... 六年级英语专题训练(连词成句30题)  1. have,playhouse,many,I,toy,i...
上班迟到情况说明英语   每个人都或多或少的迟到过那么几次,因为各种原因,可能生病,可能因为交通堵车,可能是因为天气冷,有...
小学英语教学论文 小学英语教学论文范文  引导语:英语教育一直都是每个家长所器重的,那么有关小学英语教学论文要怎么写呢...
英语口语学习必看的方法技巧 英语口语学习必看的方法技巧如何才能说流利的英语? 说外语时,我们主要应做到四件事:理解、回答、提问、...
四级英语作文选:Birth ... 四级英语作文范文选:Birth controlSince the Chinese Governmen...
金融专业英语面试自我介绍 金融专业英语面试自我介绍3篇  金融专业的学生面试时,面试官要求用英语做自我介绍该怎么说。下面是小编...
我的李老师走了四年级英语日记... 我的李老师走了四年级英语日记带翻译  我上了五个学期的小学却换了六任老师,李老师是带我们班最长的语文...
小学三年级英语日记带翻译捡玉... 小学三年级英语日记带翻译捡玉米  今天,我和妈妈去外婆家,外婆家有刚剥的`玉米棒上带有玉米籽,好大的...
七年级英语优秀教学设计 七年级英语优秀教学设计  作为一位兢兢业业的人民教师,常常要写一份优秀的教学设计,教学设计是把教学原...
我的英语老师作文 我的英语老师作文(通用21篇)  在日常生活或是工作学习中,大家都有写作文的经历,对作文很是熟悉吧,...
英语老师教学经验总结 英语老师教学经验总结(通用19篇)  总结是指社会团体、企业单位和个人对某一阶段的学习、工作或其完成...
初一英语暑假作业答案 初一英语暑假作业答案  英语练习一(基础训练)第一题1.D2.H3.E4.F5.I6.A7.J8.C...
大学生的英语演讲稿 大学生的英语演讲稿范文(精选10篇)  使用正确的写作思路书写演讲稿会更加事半功倍。在现实社会中,越...
VOA美国之音英语学习网址 VOA美国之音英语学习推荐网址 美国之音网站已经成为语言学习最重要的资源站点,在互联网上还有若干网站...
商务英语期末试卷 Part I Term Translation (20%)Section A: Translate ...