У меня есть пустая стопка с формой (1,64,96,4), где (64,96) - это пиксели, которые соответствуют изображению, снятому с экрана (кадра), а 4 - это количество кадров в стеке.Таким образом, в основном в стеке есть 4 двумерных массива (64,96).Этот стек используется как вход для сети Deep q.
Я хочу отобразить с помощью matplotlib разные кадры, но мне не удалось получить доступ к нужным элементам.
Вот код для создания стека фреймов:
frame = self.preprocess_frame(frame) # (64,96)
state = np.stack([frame] * 4, axis=-1)
state = state.reshape(1, state.shape[0], state.shape[1], state.shape[2]) # 1*64*96*4, the extra dimension is only used for Keras library
return state
Проблема заключается в том, как получить доступ к различным фреймам из стека, чтобы подать их в matplotlib.imshow
Я пробовал это, чтобы получить только один кадр, но это не сработало:
plt.imshow(state[1], state[2])
plt.show()
Заранее спасибо !!!