Skip to content
Open
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 5 additions & 3 deletions monai/losses/perceptual.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,9 +49,11 @@ class PerceptualLoss(nn.Module):

Args:
spatial_dims: number of spatial dimensions.
network_type: {``"alex"``, ``"vgg"``, ``"squeeze"``, ``"radimagenet_resnet50"``,
``"medicalnet_resnet10_23datasets"``, ``"medicalnet_resnet50_23datasets"``, ``"resnet50"``}
Specifies the network architecture to use. Defaults to ``"alex"``.
network_type: str or PercetualNetworkType Specifies the network architecture to use. Supported values are defined in
``PercetualNetworkType`` (e.g., ``"alex"``, ``"vgg"``, ``"squeeze"``,
``"radimagenet_resnet50"``, ``"medicalnet_resnet10_23datasets"``,
``"medicalnet_resnet50_23datasets"``, ``"resnet50"``).
Defaults to ``PercetualNetworkType.alex``.
is_fake_3d: if True use 2.5D approach for a 3D perceptual loss.
fake_3d_ratio: ratio of how many slices per axis are used in the 2.5D approach.
cache_dir: path to cache directory to save the pretrained network weights.
Expand Down
Loading