次のコマンドを使用して .pb ファイルを作成しました。
flow --model ../YOLOv2/alexeyAB_darknet/darknet-master/cfg/yolov2-dppedestrian.cfg --load ../YOLOv2/alexeyAB_darknet/darknet-master/backup/yolov2-dppedestrian_33900.weights --savepb
モデルは正常に作成されましたが、それを Java tensorflow アプリケーションにロードすると、次のエラーが発生します。
Exception in thread "Thread-9" org.tensorflow.TensorFlowException: Could not find meta graph def matching supplied tags: { serve }. To inspect available tag-sets in the SavedModel, please use the SavedModel CLI: saved_model_cli
問題はコードの 2 行目にあります。
String model_path = "/home/adisys/Desktop/cloudiV2/models/yolo_pedestrian/saved_model";
SavedModelBundle model = SavedModelBundle.load(model_path, "serve");
深く掘り下げてみたところ、次のリンクが見つかりました: tensorflow サービングで pb ファイルをロードできません リンクに続いて、次のコマンドを実行しました。
saved_model_cli show --dir saved_model/
出力は次のとおりです。
/home/adisys/anaconda3/lib/python3.6/site-packages/h5py/init.py:34: FutureWarning: issubdtype の 2 番目の引数の float から np.floating への変換は非推奨です。今後は np.float64 == np.dtype(float).type として扱われます。from ._conv import register_converters as _register_converters
指定された SavedModel には、次のタグセットが含まれています。
ご覧のとおり、タグセットは表示されていません。問題は何ですか?