创建将整个图像数据集转换为注释文件（.xml文件）的自动方式

Question

我有一个数据集，它已经在检测屋顶时被裁剪了。（1个图像只有1个屋顶） 例：-

我熟悉labelImg工具，但它需要时间

是否有任何方法可以将文件夹中的每个图像文件转换为注释文件，如图像大小为（xmin、ymin、xmax、ymax）并保存为PascalVOC格式的.xml文件

Answer 1

如果图像已经被裁剪，那么您应该能够通过检查图像来获得边界框

from PIL import Image

paths_of_images_in_training_directories = #recursive directory walk of training dir

for training_image in paths_of_images_in_training_directories:
    img = Image.open(training_image)
    image_w, image_h = img.size

    # Determine xmin,ymin,xmax,ymax or annotations
    xmin = 0
    ymin = 0
    xmax = image_w
    ymax = image_h

    # ...write annotations to .xml file