Update square_crop.py
Browse files- square_crop.py +1 -1
square_crop.py
CHANGED
@@ -58,7 +58,7 @@ def crop_annotation(img_path, ann, out_dir, pad_color=0):
|
|
58 |
stem = Path(img_path).stem
|
59 |
suffix = Path(img_path).suffix
|
60 |
out_name = f"{stem}_ann{ann['id']}{suffix}"
|
61 |
-
crop.save(out_dir
|
62 |
|
63 |
|
64 |
def run_square_crop(input_dir, coco_json_path, cropped_dir):
|
|
|
58 |
stem = Path(img_path).stem
|
59 |
suffix = Path(img_path).suffix
|
60 |
out_name = f"{stem}_ann{ann['id']}{suffix}"
|
61 |
+
crop.save(os.path.join(out_dir, out_name))
|
62 |
|
63 |
|
64 |
def run_square_crop(input_dir, coco_json_path, cropped_dir):
|