hrnetv2_w18_imagenet_pretrained.pth
PyTorch HRNet
在线加载模型地址

确定自己使用的框架并导入对应的库。导入库实现样例代码可参考 文档中心-预训练模型使用教程

在代码中实现加载预训练模型地址

调用模型的实现方法

import torch

from flyai.utils import remote_helper


model = HRnet()

path =  remote_helper.get_remote_date('https://www.flyai.com/m/hrnetv2_w18_imagenet_pretrained.pth')

dict = torch.load(path)

model.load_state_dict(dict)