def observe(image): im_gs = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY) observations = np.reshape(image, (-1, 3)).astype(np.float32) kernels_run = features.conv_features(im_gs) kernel_observations = np.reshape( kernels_run, (-1, kernels_run.shape[2])).astype(np.float32) all_observations = np.hstack(( observations, kernel_observations, )).astype(np.float32) return all_observations
def observe(image): im_gs = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY) observations = np.reshape(image, (-1, 3)).astype(np.float32) kernels_run = features.conv_features(im_gs) kernel_observations = np.reshape( kernels_run, (-1, kernels_run.shape[2]) ).astype(np.float32) all_observations = np.hstack( ( observations, kernel_observations, ) ).astype(np.float32) return all_observations