переупорядочить содержимое строк данных для keras cnn

Question

У меня есть датафрейм с формой (60,), т.е. 60 строк videos_df['videos'].shape : (60,)

В каждой строке разное количество изображений (64 * 64), проанализированных из папок: 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) .....

Строки, содержащие видеокадры, различаются по длине в зависимости от длины видео. Как я могу использовать эти видеокадры в кератах, где каждая строка видео, если 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)