/radimagenet-models

Unofficial support to RadImageNet pretrained models for Pytorch

Primary LanguagePythonApache License 2.0Apache-2.0

pytorch_radimagenet_models

Unofficial support to RadImageNet pretrained models for Pytorch

(Download link generated with https://sites.google.com/site/gdocs2direct/?pli=1)

To download the model use:

from radimagenet_models.models.resnet import radimagenet_resnet50
resnet = radimagenet_resnet50()

or

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

For now, the package is available with pip install -i https://test.pypi.org/simple/ radimagenet-models