You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

gen_json.py 821 B

12345678910111213141516171819202122
  1. import json
  2. import os
  3. def dump_json_from_dict(structure, file_name):
  4. with open(file_name + '.json', 'w') as file_path:
  5. json.dump(structure, file_path)
  6. if __name__ == '__main__':
  7. # iterate over directory
  8. DIRECTORY = "imagefolder"
  9. i = 0
  10. for filename in os.listdir(DIRECTORY):
  11. default_dict = {}
  12. default_dict.update(dataset='')
  13. default_dict.update(image=(os.path.join(DIRECTORY, filename)))
  14. default_dict.update(label=[1, 2])
  15. default_dict.update(_priority=0.8)
  16. default_dict.update(_embedding='sample.bin')
  17. default_dict.update(_segmented_image=(os.path.join(DIRECTORY, filename)))
  18. default_dict.update(_processed_image=(os.path.join(DIRECTORY, filename)))
  19. i = i + 1
  20. dump_json_from_dict(default_dict, 'images/'+str(i))