Пример #1
0
  def _finalize(self, interface, action_space, use_feature_units, visualize):
    game_info = self._controllers[0].game_info()
    static_data = self._controllers[0].data()
    if not self._map_name:
      self._map_name = game_info.map_name

    if game_info.options.render != interface.render:
      logging.warning(
          "Actual interface options don't match requested options:\n"
          "Requested:\n%s\n\nActual:\n%s", interface, game_info.options)

    self._features = features.Features(game_info=game_info,
                                       action_space=action_space,
                                       use_feature_units=use_feature_units)
    if visualize:
      self._renderer_human = renderer_human.RendererHuman()
      self._renderer_human.init(game_info, static_data)
    else:
      self._renderer_human = None

    self._metrics = metrics.Metrics(self._map_name)
    self._metrics.increment_instance()

    self._last_score = None
    self._total_steps = 0
    self._episode_steps = 0
    self._episode_count = 0
    self._obs = None
    self._state = environment.StepType.LAST  # Want to jump to `reset`.
    logging.info("Environment is ready on map: %s", self._map_name)
Пример #2
0
    def _finalize(self, agent_interface_formats, interfaces, visualize):
        game_info = self._parallel.run(c.game_info for c in self._controllers)
        if not self._map_name:
            self._map_name = game_info[0].map_name

        for g, interface in zip(game_info, interfaces):
            if g.options.render != interface.render:
                logging.warning(
                    "Actual interface options don't match requested options:\n"
                    "Requested:\n%s\n\nActual:\n%s", interface, g.options)

        self._features = [
            features.features_from_game_info(
                game_info=g,
                use_feature_units=agent_interface_format.use_feature_units,
                use_raw_units=agent_interface_format.use_raw_units,
                use_unit_counts=agent_interface_format.use_unit_counts,
                use_camera_position=agent_interface_format.use_camera_position,
                action_space=agent_interface_format.action_space,
                hide_specific_actions=agent_interface_format.
                hide_specific_actions) for g, agent_interface_format in zip(
                    game_info, agent_interface_formats)
        ]

        if visualize:
            static_data = self._controllers[0].data()
            self._renderer_human = renderer_human.RendererHuman()
            self._renderer_human.init(game_info[0], static_data)
        else:
            self._renderer_human = None

        self._metrics = metrics.Metrics(self._map_name)
        self._metrics.increment_instance()

        self._last_score = None
        self._total_steps = 0
        self._episode_steps = 0
        self._episode_count = 0
        self._obs = [None] * len(interfaces)
        self._agent_obs = [None] * len(interfaces)
        self._state = environment.StepType.LAST  # Want to jump to `reset`.
        logging.info("Environment is ready on map: %s", self._map_name)
Пример #3
0
    def _finalize(self, visualize):
        self._delayed_actions = [
            collections.deque() for _ in self._action_delay_fns
        ]

        if visualize:
            self._renderer_human = renderer_human.RendererHuman()
            self._renderer_human.init(self._controllers[0].game_info(),
                                      self._controllers[0].data())
        else:
            self._renderer_human = None

        self._metrics = metrics.Metrics(self._map_name)
        self._metrics.increment_instance()

        self._last_score = None
        self._total_steps = 0
        self._episode_steps = 0
        self._episode_count = 0
        self._obs = [None] * self._num_agents
        self._agent_obs = [None] * self._num_agents
        self._state = environment.StepType.LAST  # Want to jump to `reset`.
        logging.info("Environment is ready")