forked from mindspore-Ecosystem/mindspore
23 lines
821 B
Python
23 lines
821 B
Python
import json
|
|
import os
|
|
|
|
def dump_json_from_dict(structure, file_name):
|
|
with open(file_name + '.json', 'w') as file_path:
|
|
json.dump(structure, file_path)
|
|
|
|
if __name__ == '__main__':
|
|
# iterate over directory
|
|
DIRECTORY = "imagefolder"
|
|
i = 0
|
|
for filename in os.listdir(DIRECTORY):
|
|
default_dict = {}
|
|
default_dict.update(dataset='')
|
|
default_dict.update(image=(os.path.join(DIRECTORY, filename)))
|
|
default_dict.update(label=[1, 2])
|
|
default_dict.update(_priority=0.8)
|
|
default_dict.update(_embedding='sample.bin')
|
|
default_dict.update(_segmented_image=(os.path.join(DIRECTORY, filename)))
|
|
default_dict.update(_processed_image=(os.path.join(DIRECTORY, filename)))
|
|
i = i + 1
|
|
dump_json_from_dict(default_dict, 'images/'+str(i))
|