Как подготовить набор данных для CONVLSTM2D?

Question

Я пытаюсь использовать CONVLSTM2D в керасе, но я не знаю, как подготовить и получить данные тренировки. Я не знаю, являются ли входные данные одним изображением с последовательностью кадров для изучения или ячейкой изображения (как matlab).

Answer 1

Поскольку вы упомянули о кадрах, я предполагаю, что вы хотите обучать модели на видео. Для видео вы можете конвертировать видео в один numpy файл и модель поезда, как указано ниже в фрагменте кода.

>>> np.load('dataset/scene_0.npy').shape()
(1000, 3, 400, 400)
>>> np.load('dataset/category_0.npy').shape()
(6, 1, 1000, 1)

Затем вам нужно использовать генератор, который подает образец модели по образцу, используя .fit_generator () .

Подробную информацию об этом топи можно найти c здесь