Iterate over the folder, convert the yml file into a json file in batches, and pre-process the label

import os

import json

import yaml

def yml2json(yml_path, json_path):

for root, dirs, files in (yml_path):

# root indicates the folder path currently being accessed

# dirs means the subdirectory name list under this folder

# files means the file list in this folder

for idx, file_name in enumerate(files):

if file_name.find(".yml") != -1: # Find only yml format files

yml_file_path = yml_path + file_name

with open(yml_file_path, 'r') as yaml_file:

if 0: # Do you need to delete the first line (used when encountering the first line is %YAML)

with open(yml_file_path, 'r') as file:

lines = ()

with open(yml_file_path, 'w') as file:

(lines[1:])

if 1:

yaml_data = (yaml_file, )

# Convert to JSON format

json_data = (yaml_data)

file_name = file_name.split(".")[0] # Remove the suffix

json_file_path = json_path + file_name + ".json"

with open(json_file_path, 'w') as f:

(json_data)

print(idx+1, end="")

print(" yml file converted")

print(yml_file_path, end="")

print(" all files done!")

if __name__ == '__main__':

yml_path = "E:\\dataset\\annotations\\" # The folder location where the yml labeled files are stored

json_path = "E:\\dataset\\annotations_json\\" # folder locations that are expected to store json format labeled files

yml2json(yml_path, json_path)