У меня проблемы с сохранением массива numpy 4d

Question

У меня проблемы с сохранением 4d массива. Чтобы объяснить, сначала я изменил размеры изображений, используя

N = df.shape[0]
x_train = np.empty((N, 224, 224, 3), dtype=np.uint8)

for i, image_id in enumerate(tqdm(df['path'])):
x_train[i, :, :, :] = preprocess_image(
    f'{image_id}'
)

Когда я печатаю размер, я получаю (3658, 224, 224, 3). После сохранения x_train в виде файла .npy с использованием

np.save('x_train.npy', x_train)

и загрузки файла с использованием

x_train = np.load('x_train.npy')

я получаю другой размер (2926, 224,224,3) и получаю I ' Я не уверен, почему. Любые идеи?