Spaces:

MLBench
/

Inspectech_segmentation

Running

Ayesha-Majeed commited on 12 days ago

Commit

3fc284c

verified ·

1 Parent(s): f886804

Update binary_segmentation.py

Files changed (1) hide show

binary_segmentation.py CHANGED Viewed

@@ -540,7 +540,7 @@ class BinarySegmenter:
                 logger.info("✅ BiRefNet quantized to INT8")
             self.transform = transforms.Compose([
-                transforms.Resize((420, 420)),
                 transforms.ToTensor(),
                 transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
             ])

                 logger.info("✅ BiRefNet quantized to INT8")
             self.transform = transforms.Compose([
+                transforms.Resize((512, 512)),
                 transforms.ToTensor(),
                 transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
             ])