vggface_16.pth
PyTorch VGG
在线加载模型地址

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

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

调用模型的实现方法

# 官网上是t7格式,事先进行了转换,转为了pth格式 

# 由于vgg face的最后一层是2622维,因此无法利用torchvison中定义的vgg16载入所有权重,建议只载入特征提取的权重,或是重新自定义网络模型

# 预先定义的网络模型的名称与torchvison中定义的vgg16模型的名称相同

import torchvision

from flyai.utils import remote_helper

path = remote_helper.get_remote_data("https://www.flyai.com/m/vggface_16.pth")

weights = torch.load("./models/VGG_FACE_pytorch.pth")