您的位置:首页 > 其它

将darknet生成的.weight转化为.pb文件

2017-08-18 17:36 176 查看
由于想把yolo生成的模型移植到android手机上,所以需要将生成的权重文件转化为.pb格式,以下为步骤:

1.安装darkflow

去该网址https://github.com/thtrieu/darkflow.git下载darkflow源码。官方介绍说只适合python3,不过我的python2.7正常使用。

打开darkflow根目录下darkflow/utils中的loader.py,将其中121行中的self.offset由16改为20:

self.offset = 20


之后,终端在darkflow根目录下输入
pip install .
,显示
Successfully built darkflow


2.生成.pb文件

将darknet中你用到的cfg和生成的.weights文件拷贝到darkflow的根目录,然后输入一下命令

flow --model (你用到的).cfg --load (你生成的).weights --savepb


搞定,收工!
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐