python - 用零填充 3d numpy 图像数组

Question

我有 357 个 .bmp 图像(形状:(357,227,227))

我将它们读入 numpy 数组，然后将它们填充为标准大小

(4608,227,227)。问题是当我从填充的 .npy 中读取图像时，所有

图像显示为黑色，意味着所有图像都用零填充。

我不知道为什么将所有图像填充为零，我需要保留图像。以下是我尝试过的:

allfiles = os.listdir(pth_upd)
files = []
columns = ['data']
for file in allfiles:
    files.append(file) if ('.bmp' in file) else None
    samples = np.empty((1,227,227))

for file in files:
    img = cv2.imread(os.path.join(pth_upd,file),0)
    img = img.reshape(1,227,227)
    img=img.astype(np.float32)
    samples = np.append(samples, img, axis=0)

    if (len(samples)< 4608) :

        pad_size=4608-len(samples)       

        samples = np.pad(samples,(( pad_size,0),(0,0),(0,0)),mode='constant', constant_values=0) 

        f_name=format(folder)
        np.save(f_name, samples)
        print('saved')
        print(samples.shape)

    else:
        None

Answer 1

发生这种情况的原因是您在循环内对所有图像文件进行填充。

因此，每当您进行填充时，您都会覆盖在上一次迭代中加载的任何图像。

您应该在完成所有图像文件的循环之后进行填充。