Function optimize_mask

optimize_mask.py:57–61

(args)

        Image.fromarray(mask).save(os.path.join(args.save_dir, mask_name))

def optimize_mask(args):
    mask_names = os.listdir(args.generated_mask_dir)
    with ProcessPoolExecutor(max_workers=args.num_workers) as executor:
        list(tqdm(executor.map(process_mask, [mask_name for mask_name in mask_names], [args] * len(mask_names)),
                  total=len(mask_names)))

_args = parse_args()
optimize_mask(_args)