Exemplo n.º 1
0
 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)
Exemplo n.º 2
0
 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)
Exemplo n.º 3
0
 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)