lam/utils/gen_json.py

import json
import glob
import sys
import os

data_root = sys.argv[1]
save_path = sys.argv[2]

all_img_list = []
for hid in os.listdir(data_root):
    all_view_imgs_dir = os.path.join(data_root, hid, "kinect_color")
    if not os.path.exists(all_view_imgs_dir):
        continue
    
    for view_id in os.listdir(all_view_imgs_dir):
        imgs_dir = os.path.join(all_view_imgs_dir, view_id)
        for img_path in glob.glob(os.path.join(imgs_dir, "*.png")):
            all_img_list.append(img_path.replace(data_root + "/", ""))

print(f"len:{len(all_img_list)}")
print(all_img_list[:3])
with open(save_path, 'w') as fp:
    json.dump(all_img_list, fp, indent=4)