Question

我使用以下命令来创建.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

问题出在代码的第二行：

String model_path = "/home/adisys/Desktop/cloudiV2/models/yolo_pedestrian/saved_model";
SavedModelBundle model = SavedModelBundle.load(model_path, "serve");

我尝试深入研究，发现以下链接： Can not load pb file in tensorflow serving 在链接之后，我运行了以下命令：

saved_model_cli show --dir saved_model/

输出如下：

/home/adisys/anaconda3/lib/python3.6/site-packages/h5py/init.py:34：FutureWarning：issubdtype的第二个参数从float转换为np。不建议使用float。将来，它将被视为np.float64 == np.dtype（float）.type。从._conv导入register_converters作为_register_converters
给定的SavedModel包含以下标记集：

可以看出，没有显示标签集。可能是什么问题？

Answer 1

我刚刚看到了您的帖子，我确定问题已经解决，但是我将评论留给其他使用Darkflow的人。需要将命令--savepb分配为--savepb True

保存的.pb文件（使用darkflow创建）未加载到Java

1 个答案: