Skip to content

Commit e2d0a02

Browse files
committed
DCO Remediation Commit for Mohammad Amanour Rahman <[email protected]>
I, Mohammad Amanour Rahman <[email protected]>, hereby add my Signed-off-by to this commit. Signed-off-by: Mohammad Amanour Rahman <[email protected]>
1 parent 7ce29a1 commit e2d0a02

File tree

1 file changed

+10
-6
lines changed

1 file changed

+10
-6
lines changed

monai/losses/perceptual.py

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
torchvision, _ = optional_import("torchvision")
2424

2525

26-
class PercetualNetworkType(StrEnum):
26+
class PerceptualNetworkType(StrEnum):
2727
alex = "alex"
2828
vgg = "vgg"
2929
squeeze = "squeeze"
@@ -49,11 +49,15 @@ class PerceptualLoss(nn.Module):
4949
5050
Args:
5151
spatial_dims: number of spatial dimensions.
52-
network_type: str or PercetualNetworkType Specifies the network architecture to use. Supported values are defined in
53-
``PercetualNetworkType`` (e.g., ``"alex"``, ``"vgg"``, ``"squeeze"``,
54-
``"radimagenet_resnet50"``, ``"medicalnet_resnet10_23datasets"``,
55-
``"medicalnet_resnet50_23datasets"``, ``"resnet50"``).
56-
Defaults to ``PercetualNetworkType.alex``.
52+
network_type : str
53+
One of:
54+
- "alex"
55+
- "vgg"
56+
- "squeeze"
57+
- "radimagenet_resnet50"
58+
- "medicalnet_resnet10_23datasets"
59+
- "medicalnet_resnet50_23datasets"
60+
- "resnet50"
5761
is_fake_3d: if True use 2.5D approach for a 3D perceptual loss.
5862
fake_3d_ratio: ratio of how many slices per axis are used in the 2.5D approach.
5963
cache_dir: path to cache directory to save the pretrained network weights.

0 commit comments

Comments
 (0)