Skip to content

GitLab

Commit 447750d8 authored by Abdullah Rashwan's avatar Abdullah Rashwan Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 389033478
parent ff2a881b
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
official/vision/beta/tasks/maskrcnn.py
View file @ 447750d8
...@@ -261,12 +261,15 @@ class MaskRCNNTask(base_task.Task): ...@@ -261,12 +261,15 @@ class MaskRCNNTask(base_task.Task):
metrics.append(tf.keras.metrics.Mean(name, dtype=tf.float32)) metrics.append(tf.keras.metrics.Mean(name, dtype=tf.float32))
else: else:
if self._task_config.annotation_file: if (not self._task_config.model.include_mask
) or self._task_config.annotation_file:
self.coco_metric = coco_evaluator.COCOEvaluator( self.coco_metric = coco_evaluator.COCOEvaluator(
annotation_file=self._task_config.annotation_file, annotation_file=self._task_config.annotation_file,
include_mask=self._task_config.model.include_mask, include_mask=self._task_config.model.include_mask,
per_category_metrics=self._task_config.per_category_metrics) per_category_metrics=self._task_config.per_category_metrics)
else: else:
# Builds COCO-style annotation file if include_mask is True, and
# annotation_file isn't provided.
annotation_path = os.path.join(self._logging_dir, 'annotation.json') annotation_path = os.path.join(self._logging_dir, 'annotation.json')
if tf.io.gfile.exists(annotation_path): if tf.io.gfile.exists(annotation_path):
logging.info( logging.info(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment