Расширение X_train для получения формы (batch_size, width,height,colour_channels)

#tensorflow #machine-learning #conv-neural-network

#tensorflow #машинное обучение #conv-нейронная сеть

Вопрос:

Я создал сверточную модель NN для классификации музыкальных жанров в Tensorflow, и я столкнулся с проблемой:

У меня есть 999 изображений музыкальных спектрограмм в моем X, а также соответствующие метки (y). Изображения имеют размер (128,1293). Когда я проверяю форму X, она показывает мне (999,), что означает количество изображений; однако я хочу расширить X, чтобы включить размеры изображения (1 цветовой канал) до (999,128,1293,1) для подачи в мой CNN.

введите описание изображения здесь

Я где-то читал, что вы можете сделать это с помощью cv2.resize() , но я не уверен в этом методе. Любая помощь будет оценена.

Комментарии:

1. что такое тип X ? Это тензор?

2. X — серия pandas