无法在对象检测训练中加载finetune_checkpoint答案

【问题标题】：Unable to load finetune_checkpoint in object detection training无法在对象检测训练中加载finetune_checkpoint
【发布时间】：2018-01-01 16:49:10
【问题描述】：

我正在关注这个tutorial。

我正在使用类似于this 的 Oxford-IIIT Pet 数据和配置文件。

在 pipeline_config 文件中，我将微调检查点指定为

fine_tune_checkpoint: "{PATH TO}/ssd_mobilenet_v1_coco_11_06_2017/model.ckpt"
from_detection_checkpoint: false

但是当我跑步时， python object_detection/train.py ... --train_dir=${PATH_TO_TRAIN_DIR}

我收到警告列表。

警告：root:Variable [MobilenetV1/Conv2d_0/BatchNorm/beta] 在检查点中不可用。

警告：root:Variable [MobilenetV1/Conv2d_9_pointwise/weights] 在检查点不可用

然后程序退出说

ValueError：没有要保存的变量

我已经使用

检查了 python 中的 ckpt

print_tensors_in_checkpoint_file('{PATH TO}/ssd_mobilenet_v1_coco_11_06_2017/model.ckpt/model.ckpt', '',"")

输出是一个列表。

FeatureExtractor/MobilenetV1/Conv2d_0/BatchNorm/beta (DT_FLOAT) [32]

FeatureExtractor/MobilenetV1/Conv2d_9_pointwise/weights (DT_FLOAT) [1,1,512,512]

注意：当finetune_checkpoint被禁用时，训练脚本运行良好

from_detection_checkpoint: true

在管道配置文件中

【问题讨论】：

【解决方案1】：

您正在使用的检查点文件实际上需要设置from_detection_checkpoint: true，因为该模型被训练为检测器（因此没有被禁用）。

如果您想从为 Imagenet 分类预训练的 Slim models 之一开始训练，您只需设置 from_detection_checkpoint: false。

【讨论】：