创建将整个图像数据集转换为注释文件(.xml 文件)的自动方法

Create automatic way to convert whole image dataset into annotation file(.xml file)

我有一个数据集,它已经被裁剪检测屋顶。 (1 张图像只有 1 个屋顶) 前任:-

我熟悉 labelImg 工具,但需要时间。

有什么方法可以将每个图像文件从文件夹转换为注释文件,例如图像大小为 (xmin,ymin,xmax,ymax) 并保存为 PascalVOC 格式。xml 文件。

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

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