def __init__(self, config, tracking_camera, scene_type="building", gpu_count=0): BaseEnv.__init__(self, config, scene_type, tracking_camera) self.camera_x = 0 self.walk_target_x = 1e3 # kilometer away self.walk_target_y = 0 self.k = 5 self.robot_tracking_id = -1 self.scale_up = 4 self.dataset = ViewDataSet3D( transform = np.array, mist_transform = np.array, seqlen = 2, off_3d = False, train = False, overwrite_fofn=True, env = self) self.ground_ids = None if self.gui: assert(self.tracking_camera is not None) self.gpu_count = gpu_count self.nframe = 0 self.eps_reward = 0 self.reward = 0 self.eps_count = 0 self._robot_introduced = False self._scene_introduced = False
def __init__(self, config, tracking_camera, scene_type="building", gpu_idx=0): BaseEnv.__init__(self, config, scene_type, tracking_camera) self.camera_x = 0 self.walk_target_x = 1e3 # kilometer away self.walk_target_y = 0 self.k = 5 self.robot_tracking_id = -1 self.scale_up = 4 self.dataset = None self.ground_ids = None if self.gui: assert (self.tracking_camera is not None) self.gpu_idx = gpu_idx self.assign_ports() self.nframe = 0 self.eps_reward = 0 self.reward = 0 self.eps_count = 0 self._robot_introduced = False self._scene_introduced = False
def __init__(self, config, scene_type="building", gpu_count=0): BaseEnv.__init__(self, config, scene_type) ## The following properties are already instantiated inside xxx_env.py: # @self.timestep # @self.frame_skip self.camera_x = 0 self.walk_target_x = 1e3 # kilometer away self.walk_target_y = 0 self.temp_target_x = 0 # added randomness self.temp_target_y = 0 # added randomness self.k = 5 self.robot_tracking_id = -1 self.scale_up = 4 self.dataset = ViewDataSet3D(transform=np.array, mist_transform=np.array, seqlen=2, off_3d=False, train=False, overwrite_fofn=True, env=self) self.ground_ids = None if self.gui: assert (self.tracking_camera is not None) self.gpu_count = gpu_count self.nframe = 0 self.eps_reward = 0 self.reward = 0 self.eps_count = 0 self._robot_introduced = False self._scene_introduced = False