我一直在尝试使用tf.keras.layers.Reshape函数将其重塑为NCHW格式,但最终的xml文件输出如下:
重塑为(1,6,26,26)后有一个转置图层,最终形状为(1,26,6,26)我不确定为什么会有转置图层,我希望形状为(1,6,26,26)
原因是什么?
发布于 2021-02-02 16:49:36
首先,您需要了解模型网络的拓扑结构。然后,您可以重塑和批处理。
您可以使用模型优化器来操纵输入大小。你可以这样做:Model Optimizer Advanced Reshape, Batching
https://stackoverflow.com/questions/65980309
复制相似问题