| 13 | |
| 14 | |
| 15 | def parse_args(): |
| 16 | parser = argparse.ArgumentParser() |
| 17 | parser.add_argument("--area_threshold", type=int, default=20) |
| 18 | parser.add_argument("--generated_mask_dir", type=str, default="../dataset/ILGeneration/coco_semantic_light_text2img/label") |
| 19 | parser.add_argument("--save_dir", type=str, default="../dataset/ILGeneration/coco_semantic_light_text2img/optimized_label_p20") |
| 20 | parser.add_argument("--visualize", action="store_true") |
| 21 | parser.add_argument("--num_workers", type=int, default=10) |
| 22 | args = parser.parse_args() |
| 23 | |
| 24 | os.makedirs(args.save_dir, exist_ok=True) |
| 25 | |
| 26 | return args |
| 27 | |
| 28 | def process_mask(mask_name, args): |
| 29 | mask_path = os.path.join(args.generated_mask_dir, mask_name) |