TF Detect应用程序无法使用我自己的权重

时间:2017-11-10 00:45:45

标签: android object mobile detection tensorflow-gpu

我按照对象检测Android应用程序的指示,使用预训练的tiny-yolo-voc成功运行应用程序。

现在我已经使用Darknet自己使用我自己的数据集(1class)训练了另一个模型。然后我使用Darkflow命令将权重文件转换为pb类型格式

./flow --model cfg/yolo-1class.cfg --load bin/yolo-marker_1class.weights --savepb --verbalise

使用summerize_graph工具,我检查了yolo-marker_1class.pb是否有相同的'输入'和'输出'节点

我手动将此pb文件复制到资产文件夹并单击“运行”按钮后,我遇到了这个错误,当我在预训练的tiny-yolo-voc.pb中构建时没有发生这种错误 有人可以帮我吗?Error Image

0 个答案:

没有答案