Skip to content

Commit ae87ab8

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 2faba15 commit ae87ab8

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

monai/networks/nets/cell_sam_wrapper.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff 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)

0 commit comments

Comments
 (0)