| import argparse | |
| import os | |
| def main(args): | |
| os.makedirs(args.output_dir, exist_ok=True) | |
| if __name__ == '__main__': | |
| parser = argparse.ArgumentParser() | |
| parser.add_argument("input_image", type=str) | |
| parser.add_argument("T", type=int) | |
| parser.add_argument("K", type=int) | |
| parser.add_argument("model", type=str) | |
| parser.add_argument("operator_config", type=str) | |
| parser.add_argument("noise_config", type=str) | |
| parser.add_argument("--output-dir", default=".", type=str) | |
| main(parser.parse_args()) |