無法物體檢測培訓

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。