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:
instead of "stuff." Default to `None`, and it means categories are not
classified into these two categories.
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.
"""
self._pq_metric_module = panoptic_quality.PanopticQuality(
......
......@@ -174,8 +174,8 @@ def panoptic_fpn_coco() -> cfg.ExperimentConfig:
is_thing.append(True if idx <= num_thing_categories else False)
config = cfg.ExperimentConfig(
runtime=cfg.RuntimeConfig(
mixed_precision_dtype='bfloat16', enable_xla=True),
runtime=cfg.RuntimeConfig(
mixed_precision_dtype='bfloat16', enable_xla=True),
task=PanopticMaskRCNNTask(
init_checkpoint='gs://cloud-tpu-checkpoints/vision-2.0/resnet50_imagenet/ckpt-28080', # pylint: disable=line-too-long
init_checkpoint_modules=['backbone'],
......
......@@ -131,7 +131,7 @@ class PasteMasks(tf.keras.layers.Layer):
def __init__(self, output_size: List[int],
grid_sampler, **kwargs):
"""Generates panoptic segmentation masks.
"""Resizes and pastes instance masks to match image size.
Args:
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