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):
3333 auto_resize_inputs: whether to resize inputs before passing to the network.
3434 network_resize_roi: expected input size for the network.
3535 checkpoint: checkpoint file to load the SAM weights from.
36- return_features: whether to return features
36+ return_features: whether to return features
3737
3838 """
39-
39+
4040 def __init__ (
4141 self ,
4242 auto_resize_inputs = True ,
@@ -77,7 +77,7 @@ def forward(self, x):
7777 if self .auto_resize_inputs :
7878 x = F .interpolate (x , size = self .network_resize_roi , mode = "bilinear" )
7979
80- x = self .model .image_encoder (x )
80+ x = self .model .image_encoder (x )
8181
8282 if not self .return_features :
8383 x = self .model .mask_decoder (x )
You can’t perform that action at this time.
0 commit comments