def convert(image_path, annotation_path):
save_path = os.path.join(os.path.dirname(annotation_path), "xml_converted")
if not os.path.exists(save_path): os.makedirs(save_path)
for file in os.listdir(annotation_path):
if file.endswith(".txt") and file != "classes.txt":
#print("Convert", file)
annotation_no_txt = os.path.splitext(file)[0]
imagePath = image_path + "/" + annotation_no_txt + ".jpg"
image = QImage()
image.load(imagePath)
imageShape = [
image.height(),
image.width(), 1 if image.isGrayscale() else 3
]
imgFolderName = os.path.basename(annotation_path)
imgFileName = os.path.basename(imagePath)
writer = PascalVocWriter(imgFolderName,
imgFileName,
imageShape,
localImgPath=imagePath)
# Read YOLO file
txtPath = annotation_path + "/" + file
tYoloParseReader = YoloReader(txtPath, image)
shapes = tYoloParseReader.getShapes()
num_of_box = len(shapes)
for i in range(num_of_box):
label = shapes[i][0]
xmin = shapes[i][1][0][0]
ymin = shapes[i][1][0][1]
x_max = shapes[i][1][2][0]
y_max = shapes[i][1][2][1]
writer.addBndBox(xmin, ymin, x_max, y_max, label, 0)
writer.save(targetFile=save_path + "/" + annotation_no_txt +
".xml")
image.load(imagePath)
imageShape = [
image.height(),
image.width(), 1 if image.isGrayscale() else 3
]
imgFolderName = os.path.basename(imgFolderPath)
imgFileName = os.path.basename(imagePath)
writer = PascalVocWriter(imgFolderName,
imgFileName,
imageShape,
localImgPath=imagePath)
# Read YOLO file
txtPath = imgFolderPath + "/" + file
tYoloParseReader = YoloReader(txtPath, image)
shapes = tYoloParseReader.getShapes()
num_of_box = len(shapes)
for i in range(num_of_box):
label = shapes[i][0]
xmin = shapes[i][1][0][0]
ymin = shapes[i][1][0][1]
x_max = shapes[i][1][2][0]
y_max = shapes[i][1][2][1]
writer.addBndBox(xmin, ymin, x_max, y_max, label, 0)
writer.save(targetFile=imgOutputPath + "/" + annotation_no_txt +
".xml")