Files
ai-toolkit/scripts/add_mask_dataset.py