tflite 可视化

问题来源:

当我将要检查对比的两个tflite拷贝到和visualize.py同一个文件夹下面,并执行python visualize.py tflite.tflite tflite.html后遇到下图的报错:

回到源代码49行发现,原来C:\Users\iwtba\Downloads\tensorflow-master\third_party\flatbuffers下面确实没有flatc文件。

相同的问题也在github上被提到,那里也给出了大概的解决思路。解答者andrewginns 的电脑环境是ubuntu,但我是windows,所以下面我按自己的情况试着解决这个问题。

https://github.com/tensorflow/tensorflow/issues/18857

解决方法:

1.下载FlatBuffer :https://github.com/google/flatbuffers

2.安装CMake :https://cmake.org/

2.1 配置Cmake:https://jingyan.baidu.com/article/19192ad8dfa3d8e53e5707c8.html

3.解压第一步下载的zip文件,在第一级目录里打开命令行输入,

cmake -G "Visual Studio 16"

这里的16对应的是Visual Studio的版本号,根据下表对应的数字来,我的是VS2019,所以使用16

但是,紧接着我又遇到了报错

大概意思就是cmake没“看到”我的电脑装了VS2019,按照stackoverflow上的方法:

  1. Open Visual Studio
  2. Go to Tools -> Get Tools and Features
  3. In the "Workloads" tab enable "Desktop development with C++"
  4. Click Modify at the bottom right

不过,到这个时候我突然发现github上已经有人把预编译好的flatc上传上去了,我们可以直接去下载,不需要上面那些繁琐的步骤!链接:https://github.com/google/flatbuffers/releases

下载并解压好flatc.exe,把它复制到 tensorflow-master/third_party/flatbuffers 下面去,并在visualize.py里把相关代码改成:

if not os.path.exists(_BINARY):
  _BINARY = resource_loader.get_path_to_datafile(
      "C:/Users/iwtba/Downloads/tensorflow-master/third_party/flatbuffers")

注意上面路径里的斜杠是/,不是你粘贴过来的\ 。

结果搞了半天在国外一篇博文上发现,有大神做了一款模型可视化工具!链接:https://lutzroeder.github.io/netron/

真香!有了它我就可以对比我导出的tflite跟官方的tflite有哪些不一样的地方了。

最终效果:https://qiita.com/PINTO/items/d74d92ece31c5f3fd040

参考文章:https://blog.51cto.com/onebig/2122019

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值