在android跑Tensorflow官方demo跑YOLO模型出现BufferOverflowExcption

如果大家想快速的在安卓手机中跑Tensorflow的物体分类、目标跟踪等代码,可以参考我之前发过的一篇博客:手机实时目标检测——五步使用Tensorflow的Android官方demo

在玩够了官方模型之后,我想将自己的YOLO模型添加进去,在将训练出来的模型.pb文件放在asset下之后,并在安卓把需要使用的模型改为YOLO之后,出现了一个问题:BufferOverflowExcption,错误截图如下所示:
在这里插入图片描述

紧接着,程序崩溃。

在搜索了很多博客之后,总结出原因如下:

1、在原始demo下,想要移植自己的模型,不仅仅是要将.pb文件放入asset,而且还要使用bazel编译和导入必要的.so和.jar文件。

虽然很多人成功地将bazel安装在windows上,但我也尝试了很久,最终以失败告终,打算改为Linux上去跑。

2、需要注意的是,YOLO模型的输入像素(INPUT_SIZE)必须是INPUT_BLOCK_SIZE的整数倍,不然也会出现崩溃的情况。

最后,我认为有一篇博客写的挺好的,如果小伙伴们还是想在windows的环境下构建工程,可以点击以下:将tensorflow训练好的模型移植到Android (MNIST手写数字识别)

发布了42 篇原创文章 · 获赞 39 · 访问量 7万+

猜你喜欢

转载自blog.csdn.net/Hanghang_/article/details/88898810