Skip to content

GitLab

Unverified Commit 7d9f35f1 authored by srihari-humbarwadi's avatar srihari-humbarwadi
Browse files

fixed line formatting

parent facc200e
Show whitespace changes
Inline Side-by-side
Showing with 7 additions and 9 deletions
official/vision/beta/projects/panoptic_maskrcnn/tasks/panoptic_maskrcnn.py
View file @ 7d9f35f1
...@@ -209,8 +209,7 @@ class PanopticMaskRCNNTask(maskrcnn.MaskRCNNTask): ...@@ -209,8 +209,7 @@ class PanopticMaskRCNNTask(maskrcnn.MaskRCNNTask):
tf.keras.metrics.Metric]: tf.keras.metrics.Metric]:
"""Build detection metrics.""" """Build detection metrics."""
metrics = [] metrics = []
num_segmentation_classes = \ num_segmentation_classes = self.task_config.model.segmentation_model.num_classes
self.task_config.model.segmentation_model.num_classes
if training: if training:
metric_names = [ metric_names = [
'total_loss', 'total_loss',
...@@ -253,8 +252,7 @@ class PanopticMaskRCNNTask(maskrcnn.MaskRCNNTask): ...@@ -253,8 +252,7 @@ class PanopticMaskRCNNTask(maskrcnn.MaskRCNNTask):
'`include_panoptic_masks` should be set to True when computing ' '`include_panoptic_masks` should be set to True when computing '
'panoptic quality') 'panoptic quality')
pq_config = self.task_config.panoptic_quality_evaluator pq_config = self.task_config.panoptic_quality_evaluator
self.panoptic_quality_metric = \ self.panoptic_quality_metric = panoptic_quality_evaluator.PanopticQualityEvaluator(
panoptic_quality_evaluator.PanopticQualityEvaluator(
num_categories=pq_config.num_categories, num_categories=pq_config.num_categories,
ignored_label=pq_config.ignored_label, ignored_label=pq_config.ignored_label,
max_instances_per_category=pq_config.max_instances_per_category, max_instances_per_category=pq_config.max_instances_per_category,
......
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