"...resnet50_tensorflow.git" did not exist on "88a05515fcc13d6a6fb8224b5c4465414d0d53c2"
Commit 4edd2a73 authored by Srihari Humbarwadi's avatar Srihari Humbarwadi Committed by Jaeyoun Kim
Browse files

Added task definition and configs for Panoptic Mask R-CNN (#10165)

* added PanopticMaskRCNNTask config

* added tests for config

* added Panoptic MaskRCNN task

* added tests for Panoptic MaskRCNN task

* fixed typo

* set default segmentation loss weight to 1.0

* added class docstrings

* support `segmentation_decoder` checkpoint

* added `init_checkpoint_modules` description

* compute `steps_per_epoch` with `train_batch_size`

* flattened semantic_segmentation loss attributes
parent d7e2f581
......@@ -47,13 +47,11 @@ class Parser(maskrcnn.Parser):
default_factory=list)
segmentation_ignore_label: int = 255
@dataclasses.dataclass
class DataConfig(maskrcnn.DataConfig):
"""Input config for training."""
parser: Parser = Parser()
@dataclasses.dataclass
class PanopticMaskRCNN(maskrcnn.MaskRCNN):
"""Panoptic Mask R-CNN model config."""
......
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