| import os
|
| import json
|
| def find_files(directory, extensions):
|
| results = []
|
| for foldername, subfolders, filenames in os.walk(directory):
|
| for filename in filenames:
|
| if filename.endswith(extensions):
|
| file_path = os.path.abspath(os.path.join(foldername, filename))
|
| results.append(file_path)
|
| return results
|
|
|
| def count_customhumans(root):
|
| directory_path = ['CustomHumans/mesh']
|
|
|
| extensions = ('.ply', '.obj')
|
|
|
| lists = []
|
| for dataset_path in directory_path:
|
| dir = os.path.join(root, dataset_path)
|
| file_paths = find_files(dir, extensions)
|
|
|
| dataset_name = dataset_path.split('/')[0]
|
| for file_path in file_paths:
|
| lists.append(file_path.replace(root, ""))
|
| with open(f'{dataset_name}.json', 'w') as f:
|
| json.dump(lists, f, indent=4)
|
|
|
| def count_thuman21(root):
|
| directory_path = ['THuman2.1/mesh']
|
| extensions = ('.ply', '.obj')
|
| lists = []
|
| for dataset_path in directory_path:
|
| dir = os.path.join(root, dataset_path)
|
| file_paths = find_files(dir, extensions)
|
| dataset_name = dataset_path.split('/')[0]
|
| for file_path in file_paths:
|
| lists.append(file_path.replace(root, ""))
|
| with open(f'{dataset_name}.json', 'w') as f:
|
| json.dump(lists, f, indent=4)
|
|
|
| if __name__ == '__main__':
|
| root = '/data/lipeng/human_scan/'
|
|
|
| count_thuman21(root) |
