evopose2d
evopose2d copied to clipboard
Published
20 hours ago •
wmcnally
wmcnally
用tf2onnx将模型从h5转为onnx后出了点问题
使用tf2onnx API: tf2onnx.convert.from_keras() 将模型转为onnx格式后，发现模型输出的尺寸是输入的1/2，而不是1/4 用Netron查看onnx格式的模型，发现结构里有4次strides=2的卷积(3个depthwise + 1个常规)，3次ConvTranspose，但是配置文件里input_shape和output_shape却是按4倍降采样的设置......
本以为是模型转换出了问题，结果我拿Netron直接查看.h5的原版模型文件，发现确实还是4次strides=2卷积+3次转置卷积，那您这个配置文件跟实际模型就不匹配了啊......