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

fixed line formatting

parent facc200e
...@@ -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,13 +252,12 @@ class PanopticMaskRCNNTask(maskrcnn.MaskRCNNTask): ...@@ -253,13 +252,12 @@ 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, offset=pq_config.offset,
offset=pq_config.offset, is_thing=pq_config.is_thing)
is_thing=pq_config.is_thing)
return metrics return metrics
......
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