np.expand_dims(img, axis=0)
时间: 2024-05-17 13:18:24 浏览: 84
基于numpy中的expand_dims函数用法
`np.expand_dims(img, axis=0)`是将数组`img`在0轴上扩展一个新维度的函数。在这个过程中,`axis`参数指定了新维度的位置。具体来说,`axis=0`表示在数组的最前面添加一个新的维度。
这个函数通常用于将单张图片转换为批量图片,即将单张图片的形状从(高度,宽度,通道数)转换为(1,高度,宽度,通道数),以便于输入到深度学习模型中进行批量处理。
阅读全文