質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

新規登録して質問してみよう
ただいま回答率
85.50%
Python 3.x

Python 3はPythonプログラミング言語の最新バージョンであり、2008年12月3日にリリースされました。

Python

Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。

Q&A

1回答

700閲覧

Object Detection APIを使って転移学習を行いたい

samyu

総合スコア11

Python 3.x

Python 3はPythonプログラミング言語の最新バージョンであり、2008年12月3日にリリースされました。

Python

Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。

0グッド

0クリップ

投稿2018/01/29 05:31

編集2022/01/12 10:55

前提・実現したいこと

Object Detection APIの学習済みモデルを使って転移学習をしたいと思っています。

発生している問題・エラーメッセージ

自分で集めた画像を使ってモデルを作ることはできたのでこちらから取得した学習済みモデルを使って転移学習をしようと試みましたが下記のようなエラーが出ました。一通りAPIを使った転移学習の方法について調べましたが、詳しく書かれているものがなかったので質問させていただきました。

NotFoundError (see above for traceback): Key FeatureExtractor/MobilenetV1/ Conv2d_8_depthwise/BatchNorm/beta/RMSProp not found in checkpoint [[Node: save_1/RestoreV2_532 = RestoreV2[dtypes=[DT_FLOAT], _device="/job:localhost/replica:0/task:0/device:CPU:0"](_arg_save_1/Const_0_0, save_1/RestoreV2_532/tensor_names, save_1/RestoreV2_532/shape_and_slices)]]

また上記のエラー以外にも多数下記のようなエラーが出てきました

2018-01-30 : W tensorflow/core/framework/op_kernel.cc:1192] Not found: Key FeatureExtractor/MobilenetV1/Conv2d_13_pointwise_2_Conv2d _4_3x3_s2_256/weights/RMSProp_1 not found in checkpoint

試したこと

今回行った手順は次の通りです。
0. trainの中身の削除
以前学習したデータが保存されているフォルダ「train」内の全データを削除
0. 学習済みモデルのダウンロード
最初の学習に使っていたのがssd_mobilenet_v1でしたのでssd_mobilenet_v1_cocoを使うことにしました
0. ファイルの移動
ダウンロードしたtarファイルにあったcheckpointとmodel.keptの3つを学習時にデータやチェックポイントが保存されるフォルダ「train」に移動
0. ssd_mobilenet_v1.configの設定
158行あたりにある fine_tune_checkpointにmodel.ckptが存在するパスを絶対パスで記述しました。
0. 学習の開始
train.pyを使って学習を開始しました

python object_detection/train.py --pipeline_config_path=パス /ssd_mobilenet_v1.config --train_dir=パス/train --logtostderr

気になる質問をクリップする

クリップした質問は、後からいつでもMYページで確認できます。

またクリップした質問に回答があった際、通知やメールを受け取ることができます。

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

guest

回答1

0

https://github.com/tensorflow/models/issues/1836

--train_dir=パス/trainに別のモデルのチェックポイントがあればそれを削除してください。

投稿2018/01/29 06:05

mkgrei

総合スコア8560

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

samyu

2018/01/29 06:27

すいません。質問に書き忘れていました。 trainの中身は毎回全て削除してから行っています。
mkgrei

2018/01/29 07:10

ssd_mobilenet_v1.configのl.177/l.179/l.191/l.193も絶対パスで設定する必要があります。 ただ、今回のエラーとは関係ない気がします。 エラーを再現できません。 バージョンの問題でしょうか。 動画のチュートリアルがあるので、リンクをはっておきます。 https://teratail.com/questions/102236
samyu

2018/01/29 08:58

独自データだけなら学習することができていました。学習済みモデルを使った転移学習ができません。 ssd_mobilenet_v1.configは以前使用したままのため、絶対パスで設定しています。 またバージョンはtensorflow 1.4.1です。
mkgrei

2018/01/29 09:22

>ダウンロードしたzipファイルにあったcheckpointとmodel.keptの3つを学習時にデータやチェックポイントが保存されるフォルダ「train」に移動 trainの外のどこかにおいてください。 そしてそのパスをssd_mobilenet_v1.configで指定してください。
samyu

2018/01/29 15:40

行ってみましたがやはり同じエラーがでました
guest

あなたの回答

tips

太字

斜体

打ち消し線

見出し

引用テキストの挿入

コードの挿入

リンクの挿入

リストの挿入

番号リストの挿入

表の挿入

水平線の挿入

プレビュー

まだベストアンサーが選ばれていません

会員登録して回答してみよう

アカウントをお持ちの方は

15分調べてもわからないことは
teratailで質問しよう!

ただいまの回答率
85.50%

質問をまとめることで
思考を整理して素早く解決

テンプレート機能で
簡単に質問をまとめる

質問する

関連した質問