Unverified Commit 863f85e6 authored by srihari-humbarwadi's avatar srihari-humbarwadi
Browse files

fixed imports

parent 00cbcad1
...@@ -21,7 +21,6 @@ from typing import List, Optional ...@@ -21,7 +21,6 @@ from typing import List, Optional
from official.core import config_definitions as cfg from official.core import config_definitions as cfg
from official.core import exp_factory from official.core import exp_factory
from official.modeling import optimization from official.modeling import optimization
from official.vision.beta.configs import common
from official.vision.beta.configs import maskrcnn from official.vision.beta.configs import maskrcnn
from official.vision.beta.configs import semantic_segmentation from official.vision.beta.configs import semantic_segmentation
...@@ -52,14 +51,14 @@ class Parser(maskrcnn.Parser): ...@@ -52,14 +51,14 @@ class Parser(maskrcnn.Parser):
include_eval_masks: bool = True include_eval_masks: bool = True
@dataclasses.dataclass @dataclasses.dataclass
class TfExampleDecoder(common.TfExampleDecoder): class TfExampleDecoder(maskrcnn.TfExampleDecoder):
"""A simple TF Example decoder config.""" """A simple TF Example decoder config."""
# Setting this to true will enable decoding category_mask and instance_mask # Setting this to true will enable decoding category_mask and instance_mask
include_eval_masks: bool = True include_eval_masks: bool = True
@dataclasses.dataclass @dataclasses.dataclass
class DataDecoder(common.DataDecoder): class DataDecoder(maskrcnn.DataDecoder):
"""Data decoder config.""" """Data decoder config."""
simple_decoder: TfExampleDecoder = TfExampleDecoder() simple_decoder: TfExampleDecoder = TfExampleDecoder()
......
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