fixed encoder_num_channels for SAM2 (not used anyway)

juglab · Jan 13, 2025 · fc7ad05 · fc7ad05
1 parent a2314db
commit fc7ad05
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/featureforest/models/SAM2/adapter.py b/src/featureforest/models/SAM2/adapter.py
@@ -28,7 +28,7 @@ def __init__(
         self.name = name
         # we need sam2 image encoder part
         self.encoder = image_encoder
-        self.encoder_num_channels = 256
+        self.encoder_num_channels = 256 * 3
         self._set_patch_size()
         self.device = device