Spaces:

MLBench
/

Inspectech_segmentation

Running

Ayesha-Majeed commited on 12 days ago

Commit

f886804

verified ·

1 Parent(s): e3e8dfb

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((400, 400)),
                 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((420, 420)),
                 transforms.ToTensor(),
                 transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
             ])