Update binary_segmentation.py

binary_segmentation.py

@@ -508,7 +508,7 @@ class BinarySegmenter:
             )
             
             self.transform = transforms.Compose([
-                transforms.Resize((512, 512)),
+                transforms.Resize((320, 320)),
                 transforms.ToTensor(),
                 transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
             ])