Update

Remove the if logic, forcing cropping even if Raw_Cropped_to_Eye set to True
informatics-isi-edu · Jan 28, 2025 · e8ac6fd · e8ac6fd
1 parent 90fdd4e
commit e8ac6fd
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/eye_ai/eye_ai.py b/eye_ai/eye_ai.py
@@ -253,13 +253,10 @@ def create_images_directory(self, bag_path: Path, ds_bag: DatasetBag, output_dir
             out_path_dir = str(out_path_no_glaucoma) if diag == 'No Glaucoma' else str(out_path_glaucoma)
 
             if crop_to_eye:
-                if row['Annotation_Function'] != 'Raw_Cropped_to_Eye':
-                    svg_path = svg_root_path / f'Cropped_{image_rid}.svg'
-                    bbox = self.get_bounding_box(svg_path)
-                    cropped_image = image.crop(bbox)
-                    cropped_image.save(f'{out_path_dir}/Cropped_{image_rid}.JPG')
-                else:
-                    image.save(f'{out_path_dir}/Cropped_{image_rid}.JPG')
+                svg_path = svg_root_path / f'Cropped_{image_rid}.svg'
+                bbox = self.get_bounding_box(svg_path)
+                cropped_image = image.crop(bbox)
+                cropped_image.save(f'{out_path_dir}/Cropped_{image_rid}.JPG')
                 image_annot_df.loc[index, 'Cropped Filename'] = 'Cropped_' + image_file_name
             else:
                 image.save(f'{str(out_path_dir)}/{image_rid}.JPG')