Closed
Description
In monai/losses/perceptual.py
I would like to report that the model names do not correspond to the models in torch hub for the medical datasets and as a result the perceptual loss class is cumbersome to deal with.
E.g. medical_resnet50_23datasets
should become medicalnet_resnet50_23datasets
in order to find the model in torch hub.
class PercetualNetworkType(StrEnum): alex = "alex" vgg = "vgg" squeeze = "squeeze" radimagenet_resnet50 = "radimagenet_resnet50" medicalnet_resnet10_23datasets = "medicalnet_resnet10_23datasets" medical_resnet50_23datasets = "medical_resnet50_23datasets" resnet50 = "resnet50"
Metadata
Metadata
Assignees
Labels
No labels