数字图像处理中的 channels_first与channels_last

断了今生、忘了曾经 提交于 2019-11-29 23:39:47

在深度学习中,不同的框架可能对应不同的影像表达,在数据处理时应做相应的转换。

在表示一组彩色图片的问题上,Theano和Caffe使用(样本数,通道数,行或称为高,列或称为宽)通道在前的方式,称为channels_first;而TensorFlow使用(样本数,行或称为高,列或称为宽,通道数)通道在后的方式,称为channels_last。

 使用numpy的moveaxis函数

x_train=gdal_array.LoadFile(label_src+'3.tif')
x_train=np.moveaxis(x_train,0,2)   #使用这种方式转为channel last

 

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!