diff --git a/src/featureforest/models/SAM2/adapter.py b/src/featureforest/models/SAM2/adapter.py index ed14e00..571d4ef 100644 --- 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