nonebot-plugin-nailongremove/data_handle/create_to_metadata.py
2024-11-12 14:41:29 +08:00

34 lines
979 B
Python

import os
import json
num=0
def create_metadata_jsonl(root_folder, output_file):
for subdir in os.listdir(root_folder):
if not os.path.isdir(os.path.join(root_folder, subdir)):
continue
label = subdir
output_name = os.path.join(root_folder, label, output_file)
with open(output_name, 'w') as outfile:
for file in os.listdir(os.path.join(root_folder, subdir)):
if file.endswith('.jpg'):
metadata = {
"id": os.path.basename(file).split('.')[0],
"file_name": f'{file}',
"label": label
}
outfile.write(json.dumps(metadata) + "\n")
print(f"Metadata file '{output_name}' has been created successfully.")
root_folder = 'nailongClassification'
output_file = 'metadata.jsonl'
create_metadata_jsonl(root_folder, output_file)