地址链接:https://www.flyai.com/m/vggface_16.pth
查看来源:http://www.robots.ox.ac.uk/~vgg/software/vgg_face/src/vgg_face_torch.tar.gz
确定自己使用的框架并导入对应的库。导入库实现样例代码可参考 文档中心-预训练模型使用教程
在代码中实现加载预训练模型地址
# 官网上是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")