问题描述
我一直在尝试使用 tf.keras.layers.Reshape 函数重塑为 NCHW 格式,但最终的 xml 文件输出如下:
reshape为(1,6,26,26)后有一个Transpose层,最终形状为(1,26)
我不知道为什么有一个 Transpose 层,我希望形状是 (1,26)
是什么原因?
解决方法
首先,您需要了解模型网络的拓扑结构。然后你可以重塑和批处理。
您可以使用模型优化器来操纵您的输入大小。
您可以这样做:Model Optimizer Advanced Reshape,Batching