tensorflow2.x的模型输入从NHWC格式转换为NCHW格式

最新推荐文章于 2023-11-21 22:40:07 发布
最新推荐文章于 2023-11-21 22:40:07 发布
阅读量1.3k 收藏 1
点赞数
分类专栏： 深度学习--基本工具 文章标签： tensorflow python 深度学习
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/BIT_Legend/article/details/122270071
版权

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)

BIT_Legend
关注
  • 0
    点赞
  • 踩
  • 1
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
专栏目录
NCHW - NHWC - CHWN 排列
爱CV
02-19 348
TensorFlow有两种数据格式NHWC和NCHW，默认的数据格式是NHWC，可以通过参数data_format指定数据格式。这个参数规定了 input和 output Tensor 的排列方式。
深度学习中的Tensor 数据格式（N,C,H,W）
d_b_的博客
05-11 1万+
文章目录深度学习中的Tensor 数据格式（N,C,H,W）一、深度学习框架中的图像格式2、数据格式3、物理存储3.1、NCHW3.2、NHWC3.3、CHWN：StridesBlocked layout4、RGB图像数据举例5、不同框架的支持 深度学习中的Tensor 数据格式（N,C,H,W） 参考： 数据格式- 图解NCHW与NHWC数据格式 NCHW-NHWC-NC/32HW32 一、深度学习框架中的图像格式 4DTensor格式：使用4D张量描述符来定义具有4个字母的2D图像批处理的格式 数
python层和c++分别实现图像维度转置算法：
hobertony_7的专栏
09-20 1117
HWCN --> NCHW python实现： import numpy as np import tensorflow as tf data_org = np.arange(1,28,1 ) data_resh = np.reshape(data_org, [3, 3, 1, 3]) print(data_resh[1, 2, 0, 2]) data1 = np.reshape...
Tensor中数据摆放顺序NC4HW4是什么意思，只知道NCHW格式，能解释以下NC4HW4格式吗？
taoqick的专栏
12-19 1127
深度学习框架TF的feature map的数据摆放格式是CHWC，onnx格式是HCHW，最近看到格式是HC4HW4不知道是什么意思？ 作者：梁德澎 链接：https://www.zhihu.com/question/337513515/answer/768632471 来源：知乎 著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。 简单来说NC4HW4的数据排布其实就是和RGBA这种交织的数据排布类似。下面简单描述下NCHW如何转换成NC4HW4，首先batch维度就是N不变，然.
Pytorch基础-张量基本操作
最新发布
11-21 18
基于线性插值的上采样：最近邻算法（nearest）、双线性插值算法（bilinear）、双三次插值算法（bicubic）等，这是传统图像处理方法。基于深度学习的上采样（转置卷积，也叫反卷积等）Unpooling的方法（简单的补零或者扩充操作）计算效果：最近邻插值算法 < 双线性插值 < 双三次插值。计算速度：最近邻插值算法 > 双线性插值 > 双三次插值。
tensorflow: (data_format) NHWC、NCHW 区别与转换
JNing
10-03 1万+
区别NHWC [batch, in_height, in_width, in_channels] NCHW [batch, in_channels, in_height, in_width] 转换NHWC –> NCHW：import tensorflow as tfx = tf.reshape(tf.range(24), [1, 3, 4, 2]) out = tf.transpose(x
NHWC和NCHW数据排布及转换(模型部署)
@bangbang的博客
05-11 1935
其中C表示的通道数可能有多种情况，例如，RGB图片格式的通道为3通道，R（红色）、G（绿色）、B（蓝色）各占一个通道，表示图片中每个像素点都有三个通道值，每个通道值范围是[0~255]，三个通道的叠加呈现出一个像素的颜色。RGB图像还有四通道的表示，除了RGB三通道之外，还有一个alpha通道，表示透明度。在intel GPU加速的情况下，希望在访问同一个channel的像素是连续的，一般存储选用NCHW也就是输入数据格式为NCHW，这样在做CNN的时候，在访问内存的时候就是连续的了，比较方便。
tensorflow中的一些基本用法-注意事项
sdzzlove的专栏
06-28 173
1，tf中默认是 nhwc，而可以改为nchw，需要更改则需要data_format =‘NCHW’， 2，各种变换可以使用tf.transform（）做更改。 3，对于tf输出矩阵如何好看即axis的使用： 一般情况下对于axis =0，到 =3，分别表示了 n h w c的变化，如何立即： 一般一个图像是这么存放的： [[[[1,2,3],[1,2,3],[1,2,3]], [[4,5,6],[4,5,6],[4,5,6]]]] 表示2行3列，每个像素是3个channel表示。 即：nh
TensorFlow有两种数据格式NHWC和NCHW
热门推荐
weixin_37801695的博客
01-25 2万+
TensorFlow有两种数据格式NHWC和NCHW，默认的数据格式是NHWC，可以通过参数data_format指定数据格式。这个参数规定了 input Tensor 和 output Tensor 的排列方式。 1、data_format 设置为 “NHWC” 时，排列顺序为 [batch, height, width, channels] 设置为 “NCHW” 时，排列顺序为 [batch,...
Opencv中NHWC转NCHW的方法(C++版)
weixin_42904656的博客
05-06 1491
opencv nhwc转nchw c++实现方式
tflite2onnx:将TensorFlow Lite模型（* .tflite）转换为ONNX
05-03
tflite2onnx-将TensorFlow Lite模型转换为ONNX tflite2onnx将TensorFlow Lite（TFLite）模型（ *.tflite ）转换为ONNX模型（ *.onnx ），并正确处理了数据布局和量化语义（有关详细信息，请*.onnx）。 强调 如果您...
EdgeBoard嵌入式AI解决方案中NHWC数据格式的性能分析
01-19
EdgeBoard深度兼容百度大脑模型资源与工具平台（EasyDL/AI Studio），极大降低了开发验证、产品集成、科研教学、项目部署门槛，适用于安防监控、工业质检、医疗诊断、农作物生长监控、无人驾驶、无人零售等场景　...
tensorflow-tools:使用Tensorflow的Python脚本
03-14
pip install git+https://github.com/sdmonov/onnx-tensorflow@maxpool_use_pool_op_instead_delation要求convert-model-to-NWHC.py将具有NCHW数据格式的GPU优化模型转换为标准NWHC表示形式convert-onnx-to-tflite....
mobilenet-v2:“反向残差和线性瓶颈”存储库
04-26
网络架构请注意纸张有误（在第四个瓶颈层之后，输出尺寸应为 ×64代替 ×64） 输入操作员Ť C ñ s ×3 转换2d -- 32 1个2个×32 瓶颈1个16 1个1个×16 瓶颈6 24 2个2个 ×24 瓶颈6 32 3 2个 ×32 瓶颈6 64 4 2个 ×...
KerasToTensorRT:这是一个简单的演示，演示了如何在具有TensorRT集成（TFTRT）的Tensorflow上或直接在TensorRT上运行Keras模型模型，而无需调用“ freeze_graph.py”
04-30
Keras到TensorRT的例子这是在不使用情况下在具有TensorRT... 该方法支持NCHW和NHWC格式，因为Tensorflow可以处理格式问题。 $ python tftrt_example.py 演示了如何在TFTRT上运行Keras Applications ResNet50。 $ pyth
Tensorflow Keras 中input_shape引发的维度顺序冲突问题（NCHW与NHWC)
Newbie_J的博客
03-11 3695
Tensorflow Keras 中input_shape引发的维度顺序冲突问题（NCHW与NHWC) 原文链接：Tensorflow Keras 中input_shape引发的维度顺序冲突问题（NCHW与NHWC) 以tf.keras.Sequential构建卷积层为例： tf.keras.layers.Conv2D(10, 3, input_shape=(2, 9, 9),padding='s...
NCHW与NHWC格式
吴建明wujianming_110117
02-09 1954
NCHW与NHWC格式 下图很好的描述了各种不同的格式。 N代表数量， C代表channel，H代表高度，W代表宽度。 NCHW其实代表的是[W H C N]，第一个元素是000，第二个元素是沿着w方向的，即001，这样下去002 003，再接着呢就是沿着H方向，即004 005 006 007…这样到019后，沿C方向，轮到了020，之后021 022 …一直到319，然后再沿N方向。 NHWC的话以此类推，代表的是[C W H N]，第一个元素是000，第二个沿C方向，即020，040, 060…一直到
tf.nn.conv2d()卷积函数 NHWC和NCHW格式转换
哈哈哈的博客
03-23 5717
1、首先介绍tf.nn.conv2d()函数, 其函数原型： conv2( input, filter, strides, padding, use_cudnn_on_gpu=None, data_format=None, name=None ) (1) input(输入)：类型为tf.float32或tf.float64。通常指需要做...
tensorflow 中data_format＂NCHW＂　or 　＂NHWC＂
lg201601的博客
05-25 1156
Tensorflow 中　NCHW 和　NHWC的区别： 其中data_format 默认 "NHWC"，也可以手动设置为 "NCHW"。这个参数规定了 input Tensor 和 output Tensor 的排列方式。 ata_format 设置为 "NHWC" 时，排列顺序为 [B, H, W, C]；设置为 "NCHW" 时，排列顺序为 [B, C, H, W]。 以图像RGB为例，具体的效果如下： 这个怎么解释呢？ 得从矩阵的表达方式说起，[B, C, H, W] B 个（C个（Ｈ（
已经转化好的.onnx模型通道顺序是NHWC怎么转换成NCHW 的.onnx 模型 python
06-22
然后，我们将输入节点的通道顺序从NHWC修改为NCHW，并创建了新的输入张量。接着，我们将原始的输入张量从模型中删除，并将新的输入张量添加到模型中。最后，我们将新的模型保存为新的.onnx文件。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值