Unverified Commit 1edd6e86 authored by srihari-humbarwadi's avatar srihari-humbarwadi
Browse files

fixed linting issues

parent c09b41da
...@@ -23,18 +23,18 @@ class PanopticSegmentationGenerator(tf.keras.layers.Layer): ...@@ -23,18 +23,18 @@ class PanopticSegmentationGenerator(tf.keras.layers.Layer):
"""Panoptic segmentation generator layer.""" """Panoptic segmentation generator layer."""
def __init__( def __init__(
self, self,
output_size: List[int], output_size: List[int],
max_num_detections: int, max_num_detections: int,
stuff_classes_offset: int, stuff_classes_offset: int,
mask_binarize_threshold: float = 0.5, mask_binarize_threshold: float = 0.5,
score_threshold: float = 0.5, score_threshold: float = 0.5,
things_overlap_threshold: float = 0.5, things_overlap_threshold: float = 0.5,
stuff_area_threshold: float = 4096, stuff_area_threshold: float = 4096,
things_class_label: int = 1, things_class_label: int = 1,
void_class_label: int = 0, void_class_label: int = 0,
void_instance_id: int = -1, void_instance_id: int = -1,
**kwargs): **kwargs):
"""Generates panoptic segmentation masks. """Generates panoptic segmentation masks.
Args: Args:
......
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