一、yolov2.weights转为TensorFlow的pb模型
1. 下载
- https://github.com/thtrieu/darkflow.git
2. 安装
- Darkflow目录下执行:pip install -e .
3. 转换
- python ./flow --model cfg/yolo.cfg --load yolo.weights –savepb
- 转换好的模型在built_graph下
二、java中使用TensorFlow
- 参考:https://github.com/szaza/tensorflow-example-java
- 此例子中session每次使用需要重新加载,在此改为单例模型,模型只加载一次,节约时间。
- 下载地址: https://pan.baidu.com/s/1Jy4I8CpIvFCkFai-GS8jzQ 密码:9qr2
- Maven引入:
<dependency>
<groupId>org.tensorflow</groupId>
<artifactId>tensorflow</artifactId>
<version>1.6.0</version>
</dependency>
注意:
- python与java中的TensorFlow版本需一致;
- 适用TensorFlow1.6及以上,需更改anchors与cfg中一致;
- 仅使用于yolo2,其他模型需更改anchors、输入大小等。
Web项目可参考:https://github.com/szaza/tensorflow-java-examples-spring