Update prepare_annotate.py
Browse files- prepare_annotate.py +1 -1
prepare_annotate.py
CHANGED
|
@@ -30,7 +30,7 @@ for fname in tqdm(os.listdir(images_dir)):
|
|
| 30 |
mask_path = os.path.join(masks_dir, name + ".png")
|
| 31 |
|
| 32 |
if not os.path.exists(mask_path):
|
| 33 |
-
print(f"
|
| 34 |
continue
|
| 35 |
|
| 36 |
# Load image and mask
|
|
|
|
| 30 |
mask_path = os.path.join(masks_dir, name + ".png")
|
| 31 |
|
| 32 |
if not os.path.exists(mask_path):
|
| 33 |
+
print(f" No mask for {fname}, skipping...")
|
| 34 |
continue
|
| 35 |
|
| 36 |
# Load image and mask
|