reshape(-1,28,28,1)方法中参数-1与1的含义解答
numpy reshape方法参数-1与1的含义
处理ASL数据集28x28灰度图做维度重塑时,用到的代码为:
x_train = x_train.reshape(-1,28,28,1)
两个参数的具体含义:
-1:reshape方法的特殊占位参数,不需要手动指定该维度的具体数值,numpy会根据数组的总元素数量、其余几个维度传入的固定值,自动推导计算出这个位置对应的维度大小。这个位置最终对应的是训练集的样本总数,比如训练集总共有27455张图,numpy会自动算出该维度值为27455,省去手动统计样本量的步骤。要注意一次reshape操作里最多只能传入一个-1,否则会出现多解导致维度计算失败。1:代表图像的色彩通道维度值。ASL数据集的图像是单通道灰度图,没有RGB三通道,所以通道维度值设为1;把数据整理成(样本数, 图像高, 图像宽, 通道数)的四维结构,是为了适配卷积神经网络等计算机视觉模型默认的NHWC输入格式要求。如果处理的是RGB彩色图像,这个位置的参数通常设为3,对应红、绿、蓝三个色彩通道。
内容的提问来源于stack exchange,提问作者JAGADEESH UNDAVALLI
相关产品推荐
相关产品推荐

