def prepare_evaluation(self): """Preapre for evaluation.""" val_json_file = os.path.join(self._params.model_dir, 'eval_annotation_file.json') if self._params.eval.val_json_file: tf.gfile.Copy(self._params.eval.val_json_file, val_json_file) else: coco_utils.scan_and_generator_annotation_file( self._params.eval.eval_file_pattern, self._params.eval.eval_samples, include_mask=False, annotation_file=val_json_file) eval_params = params_dict.ParamsDict(self._params.eval) eval_params.override({'val_json_file': val_json_file}) self._evaluator = factory.evaluator_generator(eval_params)
def prepare_evaluation(self): """Preapre for evaluation.""" eval_params = params_dict.ParamsDict(self._params.eval) if self._params.eval.use_json_file: val_json_file = os.path.join( self._params.model_dir, "eval_annotation_file.json" ) if self._params.eval.val_json_file: tf.io.gfile.copy( self._params.eval.val_json_file, val_json_file, overwrite=True ) else: coco_utils.scan_and_generator_annotation_file( self._params.eval.eval_file_pattern, self._params.eval.eval_samples, include_mask=False, annotation_file=val_json_file, dataset_type=self._params.eval.eval_dataset_type, ) eval_params.override({"val_json_file": val_json_file}) self._evaluator = factory.evaluator_generator(eval_params)
def prepare_evaluation(self): """Preapre for evaluation.""" eval_params = params_dict.ParamsDict(self._params.eval) if self._params.eval.type == 'box_and_mask': if (not self._params.eval.use_json_file or not self._params.eval.val_json_file): raise ValueError('If `eval.type` == `box_and_mask`, ' '`eval.val_json_file` is required.') if self._params.eval.use_json_file: val_json_file = os.path.join(self._params.model_dir, 'eval_annotation_file.json') if self._params.eval.val_json_file: tf.gfile.Copy( self._params.eval.val_json_file, val_json_file, overwrite=True) else: coco_utils.scan_and_generator_annotation_file( self._params.eval.eval_file_pattern, self._params.eval.eval_samples, include_mask=False, annotation_file=val_json_file, dataset_type=self._params.eval.eval_dataset_type) eval_params.override({'val_json_file': val_json_file}) self._evaluator = factory.evaluator_generator(eval_params)