|
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) |
