Method evaluate

mogen/core/evaluation/eval_hooks.py:114–136

Evaluate the results. Args: runner (:obj:`mmcv.Runner`): The underlined training runner. results (list): Output results.

(self, runner, results)

                         **eval_kwargs)

    def evaluate(self, runner, results):
        """Evaluate the results.
        Args:
            runner (:obj:`mmcv.Runner`): The underlined training runner.
            results (list): Output results.
        """
        with tempfile.TemporaryDirectory() as tmp_dir:
            eval_res = self.dataloader.dataset.evaluate(results,
                                                        work_dir=tmp_dir,
                                                        logger=runner.logger,
                                                        **self.eval_kwargs)

        for name, val in eval_res.items():
            runner.log_buffer.output[name] = val
        runner.log_buffer.ready = True

        if self.save_best is not None:
            if self.key_indicator == 'auto':
                # infer from eval_results
                self._init_rule(self.rule, list(eval_res.keys())[0])
            return eval_res[self.key_indicator]

        return None