重新排列keras cnn的数据框行内容

Question

我有一个形状为（60，）的数据框，即60行videos_df['videos'].shape : (60,)

每行中从文件夹for i in range(len(videos_df)): print(videos_df['videos'][i].shape) (1778, 64, 64) (1560, 64, 64) (3759, 64, 64) (4243, 64, 64) .....解析的图像数量不同（64 * 64）

根据视频长度，包含视频帧的行的长度是不同的，如何在keras中使用这些视频帧，如果df每个视频行都应具有自己的类标签？

dataframe

Answer 1

您的模型需要输入的形状为(?,64,64,1)。但是输入数据的形状为(40,)，每个维度为(?,64,64)。因此，您需要np.concatenate和np.newaxis。

# shape= (?,64,64,1)
x_train = np.concatenate(x_train,axis=0)[:,:,:, np.newaxis]
y_train = np.concatenate(y_train,axis=0)
x_test = np.concatenate(x_test,axis=0)[:,:,:, np.newaxis]
y_test = np.concatenate(y_test,axis=0)