AI Label

creat_xml_fordetection.py

@@ -95,7 +95,7 @@ def make_xml(filename,size_x,size_y,boxes,labels,scores,savename):
         xmax = int(boxes[i][2])
         ymax = int(boxes[i][3])
         score = int(scores[i]*1000)
-        anno.add_pic_attr(str(labels[i]), xmin, ymin, xmax, ymax,score)
+        anno.add_pic_attr("bird", xmin, ymin, xmax, ymax,score)
     anno.savefile(savename)
 def make_xml_iccv(filename,size_x,size_y,boxes,savename):
     anno = GEN_Annotations(filename)