pad image to 512x512
for line in fp: filename = (str(int_counter)+".jpeg") line_array = line.split(" ") img = Image.open(os.path.join(root_dir, line_array[2], line_array[3])) int_x_dims = int(line_array[1]) int_y_dims = int(line_array[0]) x_pad_val_added = 512-int_x_dims y_pad_val_added = 512-int_y_dims bimg = ImageOps.expand(img, border=(0,0,x_pad_val_added,y_pad_val_added)) bimg.save(os.path.join(img_write_dir, filename)) int_counter += 1 print("Image padded succesfully!") ''' #write annotations for padded images - check if float or not on dims for line in fp: filename = (str(int_counter)+".txt") line_array = line.split(" ") line_to_write_array = ["bird", 0.00, 0, 0.00, line_array[5], line_array[6], float(line_array[5])+float(line_array[7]), float(line_array[6])+float(line_array[8]), 0.00, 0.00, 0.00, 0.00, 0.00, 0.00, 0.00] file_to_write = open(os.path.join(label_write_dir, filename), "w") print((" ".join(str(x) for x in line_to_write_array))) file_to_write.write((" ".join(str(x) for x in line_to_write_array))) int_counter += 1 print("Image label generated succesfully!") '''