修正小問題與加上設定檔機制。

This commit is contained in:
yaoming 2022-10-08 22:24:15 +08:00
parent cd83f6fedf
commit a50ea84b3d
5 changed files with 82 additions and 6 deletions

View file

@ -173,7 +173,7 @@ if __name__ == "__main__":
output_path = Path(arguments.output_dir),
output_mask_path = Path(arguments.output_mask_dir),
aligned_path = Path(arguments.aligned_dir) if arguments.aligned_dir is not None else None,
force_gpu_idxs = arguments.force_gpu_idxs,
force_gpu_idxs = [ int(x) for x in arguments.force_gpu_idxs.split(',') ] if arguments.force_gpu_idxs is not None else None,
cpu_only = arguments.cpu_only)
p = subparsers.add_parser( "merge", help="Merger")