def make_rgb_arrays(self, state):
obs = state['pw_obs']
rgb = ind_to_rgb(obs, LUT)
# TODO: better size control
size = [10 * rgb.shape[0], 10 * rgb.shape[1]]
rgb = better_imresize(rgb, size, interp='nearest')
return [rgb]
def make_rgb_arrays(self, state):
obs = self.make_observation(state)
rgb = ind_to_rgb(obs, LUT)
# max_dim = float(max(rgb.shape)) #get the resize factor
# #f = max(self._render_size)/max_dim #resize for viewing
# f = max([x/float(y) for x,y in zip(self._render_size, rgb.shape)])
# size = [2*int(rgb.shape[0]*f/2.), 2*int(rgb.shape[1]*f/2.)]
# TODO: better size control
size = [10 * rgb.shape[0], 10 * rgb.shape[1]]
rgb = better_imresize(rgb, size, interp='nearest')
return [rgb]
def make_rgb_arrays(self, state):
if state['signal'] == self.positive_signal:
color = 9
elif state['signal'] == self.negative_signal:
color = 2
else:
color = 7
ind = np.array([[color]])
rgb = ind_to_rgb(ind, LUT)
size = [50, 50]
rgb = better_imresize(rgb, size, interp='nearest')
return [rgb]
def make_rgb_arrays(self, state):
rgb = np.dstack([self.im] * 3)
size = [10 * rgb.shape[0], 10 * rgb.shape[1]]
rgb = better_imresize(rgb, size, interp='nearest')
return [rgb]