#print "annotation" #print ann_food bbox = ann_food['bbox'] catId = ann_food['category_id'] cat = coco.loadCats(catId)[0] left_x = bbox[0] left_y = bbox[1] right_x = left_x + bbox[2] right_y = left_y + bbox[3] food_type = cat['name'] last_real_name = img_annotation_jpg_name.split('/') if food_type == 'hot dog': food_type = 'hotdog' ## must be a word, coco's dataset is not right write_to_file(last_real_name[-1], food_type, img_annotation_xml_name, str(img_width), str(img_height), str(left_x), str(left_y), str(right_x), str(right_y)) file.close() ### ?????? # Generat the configuration for ImageSet img = coco.loadImgs(foodImageId[5])[0] img_name = '%s/images/%s/%s'%(dataDir,dataType,img['file_name']) I = io.imread(img_name) #plt.figure() #plt.imshow(I) #plt.show() # JUST FOR DEBUGGING
elif food_type == 'hot dog': shutil.copy(img_annotation_jpg_name, "./hotdog/") elif food_type == 'sandwich': shutil.copy(img_annotation_jpg_name, "./sandwich/") elif food_type == 'carrot': shutil.copy(img_annotation_jpg_name, "./carrot/") elif food_type == 'apple': shutil.copy(img_annotation_jpg_name, "./apple/") elif food_type == 'orange': shutil.copy(img_annotation_jpg_name, "./orange/") elif food_type == 'banana': shutil.copy(img_annotation_jpg_name, "./banana/") elif food_type == 'pizza': shutil.copy(img_annotation_jpg_name, "./pizza/") write_to_file(img_annotation_jpg_name, food_type, img_annotation_xml_name, str(img_width), str(img_height), str(left_x), str(left_y), str(right_x), str(right_y)) file.close() ### ?????? # Generat the configuration for ImageSet img = coco.loadImgs(foodImageId[5])[0] img_name = '%s/images/%s/%s'%(dataDir,dataType,img['file_name']) I = io.imread(img_name) #plt.figure() #plt.imshow(I) #plt.show() # JUST FOR DEBUGGING
#!/usr/bin/env python from write_xml import write_to_file # ''' This is just for single test of one function write_to_file # Because I found that the generated xml is not compatiable with # original xml, containing the head `<?xml xxx>`, this may harm # r-cnn code and it crashes when finishing parsing #''' write_to_file("aaa.xml", "orange", "stest.xml", "480", "640", "0", "0", "123", "123")