0

Tensorflow For Poets(link)の手順に従ってテンソルフローを使用して自分のデータセットを訓練し、非常に良い予測結果を得ました。 py、label_images.py、retrained_graph.pb、retrained_labels.txtファイル。私は自分のモデルグラフとラベルファイルを持っています。これらのattrsがiosにあるOp DecodeJpegをサポートするためにOpKernelが登録されていません

その後、私はgithubからiOSのTensorflow Example Projectをダウンロードしました。プロジェクトはiOSプラットフォームで正常に動作します。私は例のモデルとラベルファイルの代わりに、私自身のretrained_graph.pb、retrained_labels.txtファイルを使用していたときにしかし、私は次のようなエラーを取得しています -

エラー:

TensorFlowグラフを作成できませんでした:無効な引数を: OpKernelは、これらのattrsでOp 'DecodeJpeg'をサポートするために登録されていませんでした。登録されたデバイス:[CPU]、登録されたカーネル:

[[ノード:DecodeJpeg = DecodeJpegacceptable_fraction = 1、チャネル= 3、dct_method = ""、fancy_upscaling =真、比= 1、try_recover_truncated = FALSE]

and

モデルをロードできませんでした:無効な引数:これらのattrsでOp 'DecodeJpeg'をサポートするOpKernelは登録されていません。登録されたデバイス:[CPU]、登録されたカーネル:

[[ノード:DecodeJpeg = DecodeJpegacceptable_fraction = 1、チャネル= 3、dct_method = ""、fancy_upscaling =真、比= 1、try_recover_truncated = FALSE]

私を助けてくれますか?私は長い間それを探してきました。 ありがとう

答えて

0

このオペレーションはCPU上でのみ実行可能ですが、イメージデコード操作はGPUに配置しています。 documentation

から

あなたは自動的に指定されたものが存在しない場合には、あなたが設定オプションにTrueにallow_soft_placementを設定できる操作を実行するための既存およびサポートされているデバイスを選択するTensorFlowをご希望の場合セッションを作成するとき。

ので、あなたは、単に

tf.Session(config=tf.ConfigProto(
    allow_soft_placement=True)) 
を使用してセッションの初期化を変更する必要が
関連する問題