地址链接:https://www.flyai.com/m/HR18-COFW.pth
查看来源:https://github.com/HRNet/HRNet-Facial-Landmark-Detection
确定自己使用的框架并导入对应的库。导入库实现样例代码可参考 文档中心-预训练模型使用教程
在代码中实现加载预训练模型地址
state = torch.load("HR18-COFW.pth")
model.load_state_dict(state)
立刻学会使用预训练模型!
hrnetv2_w32_imagenet_pretrained.pth
hrnetv2_w30_imagenet_pretrained.pth
hrnetv2_w18_imagenet_pretrained.pth
HR18-COFW.pth