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!")
'''

  

相关文章:

  • 2022-12-23
  • 2021-12-02
  • 2021-07-12
  • 2022-03-02
  • 2021-08-07
  • 2021-09-18
  • 2021-04-13
猜你喜欢
  • 2022-12-23
  • 2022-01-04
  • 2022-12-23
  • 2021-11-13
  • 2022-12-23
  • 2021-09-02
  • 2022-12-23
相关资源
相似解决方案