1
0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +01:00

allow visualizing the observation + bug fixes to coach summary

This commit is contained in:
Itai Caspi
2018-02-13 18:47:24 +02:00
committed by Itai Caspi
parent 5d1a2bc392
commit 55c8c87afc
4 changed files with 14 additions and 2 deletions

View File

@@ -41,6 +41,8 @@ class Renderer(object):
:return: None
"""
if self.is_open:
if len(image.shape) == 2:
image = np.stack([image] * 3)
if len(image.shape) == 3:
if image.shape[0] == 3 or image.shape[0] == 1:
image = np.transpose(image, (1, 2, 0))