Unverified Commit 0bee3f54 authored by srihari-humbarwadi's avatar srihari-humbarwadi
Browse files

fixed linting and docstrings

parent 9680e09d
...@@ -79,7 +79,7 @@ class PanopticQualityEvaluator: ...@@ -79,7 +79,7 @@ class PanopticQualityEvaluator:
instead of "stuff." Default to `None`, and it means categories are not instead of "stuff." Default to `None`, and it means categories are not
classified into these two categories. classified into these two categories.
rescale_predictions: `bool`, whether to scale back prediction to original rescale_predictions: `bool`, whether to scale back prediction to original
image sizes. If True, y_true['image_info'] is used to rescale image sizes. If True, groundtruths['image_info'] is used to rescale
predictions. predictions.
""" """
self._pq_metric_module = panoptic_quality.PanopticQuality( self._pq_metric_module = panoptic_quality.PanopticQuality(
......
...@@ -131,7 +131,7 @@ class PasteMasks(tf.keras.layers.Layer): ...@@ -131,7 +131,7 @@ class PasteMasks(tf.keras.layers.Layer):
def __init__(self, output_size: List[int], def __init__(self, output_size: List[int],
grid_sampler, **kwargs): grid_sampler, **kwargs):
"""Generates panoptic segmentation masks. """Resizes and pastes instance masks to match image size.
Args: Args:
output_size: A `List` of integers that represent the height and width of output_size: A `List` of integers that represent the height and width of
......
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