mujoco_env.py 文件源码

def render(self, close=False):
        if close:
            self.stop_viewer()
        else:
            #self.get_viewer().render()
            self.get_viewer().loop_once()
            data, width, height = self.get_viewer().get_image()
            return np.fromstring(data, dtype='uint8').reshape(height, width, 3)[::-1, :, :]
        return None