本文主要是介绍openmmlab加载自训练权重,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
openmmlab加载自训练权重
在openmmlab中要加载自训练的模型权重,用于自己其他数据集训练的预训练模型。只需要在config文件中添加模型初始化。
在config.py文件中的model配置中初始化。
model = dict(type='TopdownPoseEstimator',data_preprocessor=dict(),backbone=dict(),head=dict(),test_cfg=dictionary(),init_cfg=dict(type='Pretrained',checkpoint='path/best_AUC_epoch_200.pth',),
)
参考:加载自训练权重
这篇关于openmmlab加载自训练权重的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!