SegNet转ONNX后分割输出近似二值化问题咨询

转换操作说明 #

当前持有一个训练完成的SegNet分割模型，使用如下最简Python代码将Keras格式模型转换为ONNX格式：

onnx_model, _ = tf2onnx.convert.from_keras(
    model,
    [tf.TensorSpec([1, 125, 500, 1], tf.float32)],
    output_path=out_path
)

异常表现 #

• 转换得到的ONNX模型可正常加载运行，但输出结果为近似二值图像
• 已完成数值拉伸类后处理问题排查，确认ONNX模型的原生输出值本身就落在0-1区间，将输出值乘以255后得到的可视化结果与原模型输出存在明显差异：效果对比图中上方为原SegNet模型的正常输出，下方为ONNX模型的异常输出
• 已附上通过Netron查看两个模型输出端得到的网络结构截图，可用于排查模型结构差异

核心诉求 #

定位该输出异常问题的产生原因。

内容的提问来源于stack exchange，提问作者ephraim