CenterNet MobileNetV2 FPN模型pipeline.config无fine_tune_checkpoint字段问题咨询
问题解答
问题1:是否可以手动添加fine_tune_checkpoint字段完成微调配置?
可以,完全支持手动添加该字段实现预训练权重加载,配置时注意以下几点:
- 字段需放在
train_config配置块内,字段值填写你本地解压后的预训练 checkpoint 路径前缀(不需要带.data-00000-of-00001、.index这类文件后缀,例如解压后的权重文件为ckpt-0.index、ckpt-0.data-00000-of-00001,路径直接填"/你的本地路径/ckpt-0"即可) - 需配套添加
fine_tune_checkpoint_type: "detection"字段,指定加载的是检测类预训练权重 - 如果你修改了模型头的输出类别数,可额外添加
load_all_detection_checkpoint_vars: true配置,跳过维度不匹配的层的加载,避免权重加载时报错
问题2:该字段缺失是否是由于本模型输出为关键点导致?
不是,该字段缺失和模型是否支持关键点输出没有关联。
TensorFlow 官方模型动物园提供的所有预训练检测模型的默认pipeline.config,默认都不会携带fine_tune_checkpoint字段:官方默认给出的配置是适配推理或从零训练场景的,不会提前写死微调权重的本地路径,避免用户直接运行配置时出现路径不存在的报错,该字段本身就是需要用户微调时根据自己的本地文件路径手动添加的。
内容的提问来源于stack exchange,提问作者Tarun Mishra
相关产品推荐
相关产品推荐

