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',),
)

参考：加载自训练权重