本篇內容主要講解“怎么合并PASCAL VOC的目標類別”,感興趣的朋友不妨來看看。本文介紹的方法操作簡單快捷,實用性強。下面就讓小編來帶大家學習“怎么合并PASCAL VOC的目標類別”吧!
我們遇到數據集中需要將car、bus、truck合并成car,或將person、rider、pedestrain合并為person。使用修改標簽的方法就可以實現。
import os import xml.etree.ElementTree as ET #程序功能:批量修改VOC數據集中xml標簽文件的標簽名稱 def changelabelname(inputpath): listdir = os.listdir(inputpath) for file in listdir: if file.endswith('xml'): file = os.path.join(inputpath,file) tree = ET.parse(file) root = tree.getroot() for object1 in root.findall('object'): for sku in object1.findall('name'): #查找需要修改的名稱 if (sku.text == 'preName'): #‘preName’為修改前的名稱 sku.text = 'TESTNAME' #‘TESTNAME’為修改后的名稱 tree.write(file,encoding='utf-8') #寫進原始的xml文件并避免原始xml中文字符亂碼 else: pass else: pass if __name__ == '__main__': inputpath = 'E:/Research/Dataset/Test/Annotations' #此處替換為自己的路徑
到此,相信大家對“怎么合并PASCAL VOC的目標類別”有了更深的了解,不妨來實際操作一番吧!這里是億速云網站,更多相關內容可以進入相關頻道進行查詢,關注我們,繼續學習!
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。