Warvito / MedicalNet-models

Unofficial repository to easily access MedicalNet models

MedicalNet Models

Unofficial repository to easily access MedicalNet models (Med3D: Transfer Learning for 3D Medical Image Analysis). In this repository we create entrypoints to make it easy to download MedicalNet models using torch Hub.

Note: Resnet18 and Resnet34 not available since state_dict from available .pth files does not match with the ones from the ResNet class from the original implementation.

Download pretrained model

Example of code to download pretrained model:

import torch
model = torch.hub.load("Warvito/MedicalNet-models", 'medicalnet_resnet10')

Models available:

"medicalnet_resnet10"
"medicalnet_resnet10_23datasets"
"medicalnet_resnet50"
"medicalnet_resnet50_23datasets"
"medicalnet_resnet101"
"medicalnet_resnet152"
"medicalnet_resnet200"

Citing MedicalNet

If you use this code or pre-trained models, please cite the following:

    @article{chen2019med3d,
        title={Med3D: Transfer Learning for 3D Medical Image Analysis},
        author={Chen, Sihong and Ma, Kai and Zheng, Yefeng},
        journal={arXiv preprint arXiv:1904.00625},
        year={2019}
    }