2017-07-26 5 views
3

私はこれに従いますtutorialオブジェクト検出トレーニングでfinetune_checkpointをロードできません

私はOxford-IIIT Petデータとthisに似た設定ファイルを使用しています。私が実行したときにpipeline_configファイルで

、私は のpython object_detection/train.py ... --train_dir = $ {} PATH_TO_TRAIN_DIR

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

よう微調整のチェックポイントを指定している。しかし

私は警告のリストを取得しています。

警告:ルート:変数[MobilenetV1/Conv2d_0/BatchNorm/beta]はチェックポイントでは使用できません。

WARNING:ルート:チェックポイントでは使用できません変数[MobilenetV1/Conv2d_9_p​​ointwise /ウェイト]

とValueError言ってプログラムを終了:私が検査している

を保存するには、no変数を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 /ベータ(DT_FLOAT)[32]

FeatureExtractor/MobilenetV1/Conv2d_9_p​​ointwise /ウェイト(DT_FLOAT)[1,1,512,512]

:トレーニングスクリプトfinetune_checkpointがパイプラインの設定ファイルに

from_detection_checkpoint: true 

で無効にされたときに正常に実行さ

答えて

3

実際に使用しているチェックポイントファイルには、from_detection_checkpoint: trueの設定が必要です。これは、そのモード私は検出器として訓練されました(何も無効になっていません)。

Imagenet分類用に事前にトレーニングされたSlim modelsの1つからトレーニングを開始する場合は、from_detection_checkpoint: falseと設定します。

関連する問題