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',),
)
参考:加载自训练权重