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:
@@ -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))
|
||||
|
||||
Reference in New Issue
Block a user