Пример #1
0
    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
Пример #2
0
    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
Пример #3
0
    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