无法物体检测培训

Question

加载finetune_checkpoint我下面这个tutorial。

我使用类似于this牛津IIIT PET数据和配置文件。

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

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

但是当我运行， 蟒蛇object_detection/train.py ... --train_dir = $ {} PATH_TO_TRAIN_DIR

我收到警告列表。

警告：根：变量[MobilenetV1/Conv2d_0/BatchNorm /β-在检查点不可用。

。

。

。

警告：根：变量[MobilenetV1/Conv2d_9_p​​ointwise /权重]没有检查点

和程序可用的退出说

ValueError异常：无变量保存

我已检查了ckpt in python使用

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

输出是一个列表。

FeatureExtractor/MobilenetV1/Conv2d_0/BatchNorm /β（DT_FLOAT）[32]

。

。

。

FeatureExtractor/MobilenetV1/Conv2d_9_p​​ointwise /重量（DT_FLOAT）[1,1,512,512]

注意：当finetune_checkpoint由

from_detection_checkpoint: true 

在管道配置文件中禁用

培训脚本运行良好

Answer 1

设置from_detection_checkpoint: true实际需要为您所使用，因为模式的检查点文件我被训练为一个探测器（所以没有被禁用）。

如果您想从Slim models之一开始进行培训以进行Imagenet分类，您只需设置from_detection_checkpoint: false。