Initial commit

This commit is contained in:
Lihe Yang
2024-01-22 11:28:16 +08:00
committed by GitHub
parent cc5fa04cff
commit 255150aa0d

1
run.py
View File

@@ -62,6 +62,7 @@ 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: