Tensor on Android学习笔记(二) ---将自己训练的YOLOv3模型导入TenserFlow

v2版本的使用darkflow转换后移植就可以了

https://github.com/thtrieu/darkflow


但是我们是用Yolov3版本训练的模型,这个时候就遇到了问题,目前想曲线救国,但是最终没有成功

思路是通过keras做中转,先转换为.h5的模型,在转为.pb

先用keras-yolov3将.weights转为.h5 https://github.com/qqwweee/keras-yolo3

再用keras_to_tensorflow转为.pb https://github.com/amir-abdi/keras_to_tensorflow


最终效果,并不能正确识别结果,但是模型确实是可以读取的。

因为博主刚接触到机器学习,对几种框架还不是很理解,目前没有办法进一步改进,需要先去学习一段时间。如果有读者有更深入的解决方案,希望可以指导一下。

你可能感兴趣的:(Android,TensorFlow,目标检测,深度学习,YOLO)