当前位置: 首页 > 工具软件 > Core Lite > 使用案例 >

tflite模型测试

史弘致
2023-12-01

tflite模型和pb一样都需要测试,仅仅转换成功只是其中的一步,如果测试结果很差劲,就有可能转换过程中有问题,这个问题就多了,比如模型问题、有些Op问题、UNIT8/FLOAT问题,不可预知。无论哪个环节出现问题,都有可能需要重头再来!!比如换模型,后果不堪设想。

我之前提到过,这里也再说一次,目前tf RNN尚不能直接转为tflite,只有按照官方给的trick才可以,GRU是死都不能转的。如果想要所有的RNN都能顺利转为tflite,抱歉你再等一年,人家开发开发。

但这里给出其他策略/解决方案:1-换CNN的模型,很多CNN衍生的模型基本上都支持;2-采用mobilenet,其实这个也是CNN的模型,捂脸。哈哈哈

回归正题,测试的方法很显然肯定有python直接测试,那么我先试试效果如何,由于我的模型是中间模型,不是训练完的结果,效果差是可想而知的。作为对比,稍后我将测试最终pb的效果,然后再测试其tflite。

1-首先测试第一步:模型的输入输出的数据格式

打印结果如下:注意,所有的函数保持与train.py,freeze.py时的一致,环境也要一致,不要瞎几把改,自作主张的前提是你已经足够牛逼,翅膀硬了当然可以飞,啥都不会就想跑,你傻逼啊。

[{'name': 'x_mixed',
 类似资料: