Initial commit

This commit is contained in:
Lihe Yang
2024-01-22 11:55:50 +08:00
committed by GitHub
parent 255150aa0d
commit c95c9e7729

2
run.py
View File

@@ -62,11 +62,11 @@ if __name__ == '__main__':
if args.img_path.endswith('txt'):
with open(args.img_path, 'r') as f:
filenames = f.read().splitlines()
filenames = [os.path.join(args.img_path, filename) for filename in filenames]
else:
filenames = [args.img_path]
else:
filenames = os.listdir(args.img_path)
filenames = [os.path.join(args.img_path, filename) for filename in filenames]
filenames.sort()
for filename in tqdm(filenames):