Skip to content

Cannot fing the medical_resnet50 model when using perceptual loss #7645

Closed
@vasl12

Description

@vasl12

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

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions