从无法访问的项目Fork
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
#-------------------------------------# # mAP所需文件计算代码 # 具体教程请查看Bilibili # Bubbliiiing #-------------------------------------# import sys import os import glob import xml.etree.ElementTree as ET image_ids = open('VOCdevkit/VOC2007/ImageSets/Main/test.txt').read().strip().split() if not os.path.exists("./input"): os.makedirs("./input") if not os.path.exists("./input/ground-truth"): os.makedirs("./input/ground-truth") for image_id in image_ids: with open("./input/ground-truth/"+image_id+".txt", "w") as new_f: root = ET.parse("VOCdevkit/VOC2007/Annotations/"+image_id+".xml").getroot() for obj in root.findall('object'): if obj.find('difficult')!=None: difficult = obj.find('difficult').text if int(difficult)==1: continue obj_name = obj.find('name').text bndbox = obj.find('bndbox') left = bndbox.find('xmin').text top = bndbox.find('ymin').text right = bndbox.find('xmax').text bottom = bndbox.find('ymax').text new_f.write("%s %s %s %s %s\n" % (obj_name, left, top, right, bottom)) print("Conversion completed!")