技术栈
keras .h5 onnx
nodototao
8 个月前
keras .h5 onnx
.h5转onnx,获取onnx中间层特征值
因为近些年工作中基本都是使用pytorch框架,tensorflow框架还是最初的时候使用过一段时间,现在一个项目中给到的训练后的模型是基于tensorflow keras保存的.h5文件。但是在后面部署的时候需要使用onnx进行,所以需要进行一下模型的转换。