#1lanhao回复keras的h5模型很多情况下会出现控制流,控制流会造成模型融合时发生分段的情况,因此在发生分段时需要检查是否有控制流影响。当出现这种情况可以使用const去替换该节点
展开 感谢回复:
训练后保存的是pb模型,不是h5;环境变量设置了;
手册提示:如果使用的是Keras接口,在生成量化参数文件时需要在最后添加以下代码。这是由于量化参数文件是在Session对象析构前保存的,但是Keras内部的Session并不会进行自动析构,因此需要手动析构。
from tensorflow.python.keras import backend
session = backend.get_session()
session.close()
这个我也设置了,还是不行,压缩包是整个工程,方便的话您可以调试下试试,确实是有问题的;