Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci

Signed-off-by: am <am>
  • Loading branch information
pre-commit-ci[bot] authored and am committed Aug 2, 2024
1 parent d7b910f commit 51d63c9
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions monai/networks/nets/cell_sam_wrapper.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,10 +33,10 @@ class CellSamWrapper(torch.nn.Module):
auto_resize_inputs: whether to resize inputs before passing to the network.
network_resize_roi: expected input size for the network.
checkpoint: checkpoint file to load the SAM weights from.
return_features: whether to return features
return_features: whether to return features
"""

def __init__(
self,
auto_resize_inputs=True,
Expand Down Expand Up @@ -77,7 +77,7 @@ def forward(self, x):
if self.auto_resize_inputs:
x = F.interpolate(x, size=self.network_resize_roi, mode="bilinear")

x = self.model.image_encoder(x)
x = self.model.image_encoder(x)

if not self.return_features:
x = self.model.mask_decoder(x)
Expand Down

0 comments on commit 51d63c9

Please sign in to comment.