概述
tensorflow2.x的模型输入默认格式是NHWC格式，但是某些应用场景下需要在模型中将NHWC输入格式转换为输入NCHW格式，操作代码如下
import tensorflow as tf
model_path = "./xxx.h5"
output_path = "./yyy.h5"
model.load_model(model_path) # 当前输入尺寸是128*128
new_model = tf.keras.models.Sequential([Input((3, 128, 128)), tf.keras.layers.Lambda(lambda x: tf.transpose(x,[0,2,3,1])), model])
new_model.summary()
new_model.save(output_path)最后
以上就是花痴水蜜桃为你收集整理的tensorflow2.x的模型输入从NHWC格式转换为NCHW格式的全部内容，希望文章能够帮你解决tensorflow2.x的模型输入从NHWC格式转换为NCHW格式所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复