File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -33,10 +33,10 @@ class CellSamWrapper(torch.nn.Module):
33
33
auto_resize_inputs: whether to resize inputs before passing to the network.
34
34
network_resize_roi: expected input size for the network.
35
35
checkpoint: checkpoint file to load the SAM weights from.
36
- return_features: whether to return features
36
+ return_features: whether to return features
37
37
38
38
"""
39
-
39
+
40
40
def __init__ (
41
41
self ,
42
42
auto_resize_inputs = True ,
@@ -77,7 +77,7 @@ def forward(self, x):
77
77
if self .auto_resize_inputs :
78
78
x = F .interpolate (x , size = self .network_resize_roi , mode = "bilinear" )
79
79
80
- x = self .model .image_encoder (x )
80
+ x = self .model .image_encoder (x )
81
81
82
82
if not self .return_features :
83
83
x = self .model .mask_decoder (x )
You can’t perform that action at this time.
0 commit comments