quantumiracle-git/OpenBiDexHand

1import json2 3from os import listdir4from os.path import isfile, join, isdir5mypath = './'6onlyfolders = [f for f in listdir(mypath) if isdir(join(mypath, f))]7 8idx = 09data_info = {'idx': [], 'env': [], 'name': []}10for folder in onlyfolders:11    print(folder)12    fs = [join(mypath, folder, f) for f in listdir(join(mypath, folder)) if isfile(join(mypath, folder, f))]13    # print(fs)14    for f in fs:15        if f.endswith(".gif"):16            idx += 117            data_info['idx'].append(idx)18            data_info['env'].append(folder)19            data_info['name'].append(f.split('/')[-1])20 21with open('data_info.json', 'w') as f:22    json.dump(data_info, f)23    24with open('data_info.json', 'r') as f:25    data_info = json.load(f)26print(data_info)27