Commit db23c29d authored by Abdullah Rashwan's avatar Abdullah Rashwan Committed by A. Unique TensorFlower
Browse files

Internal change

PiperOrigin-RevId: 473778060
parent 9a03ed84
...@@ -15,8 +15,8 @@ ...@@ -15,8 +15,8 @@
"""All necessary imports for registration.""" """All necessary imports for registration."""
# pylint: disable=unused-import # pylint: disable=unused-import
from official.projects.centernet.configs import centernet
from official.projects.centernet.modeling import centernet_model
from official.projects.centernet.modeling.backbones import hourglass
from official.projects.centernet.tasks import centernet as centernet_task
from official.vision import registry_imports from official.vision import registry_imports
from official.vision.beta.projects.centernet.configs import centernet
from official.vision.beta.projects.centernet.modeling import centernet_model
from official.vision.beta.projects.centernet.modeling.backbones import hourglass
from official.vision.beta.projects.centernet.tasks import centernet as centernet_task
...@@ -22,7 +22,7 @@ from official.core import config_definitions as cfg ...@@ -22,7 +22,7 @@ from official.core import config_definitions as cfg
from official.core import exp_factory from official.core import exp_factory
from official.modeling import hyperparams from official.modeling import hyperparams
from official.modeling import optimization from official.modeling import optimization
from official.vision.beta.projects.centernet.configs import backbones from official.projects.centernet.configs import backbones
from official.vision.configs import common from official.vision.configs import common
......
...@@ -19,8 +19,8 @@ import tensorflow as tf ...@@ -19,8 +19,8 @@ import tensorflow as tf
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.vision.beta.projects.centernet.common import registry_imports # pylint: disable=unused-import from official.projects.centernet.common import registry_imports # pylint: disable=unused-import
from official.vision.beta.projects.centernet.configs import centernet as exp_cfg from official.projects.centernet.configs import centernet as exp_cfg
class CenterNetConfigTest(tf.test.TestCase, parameterized.TestCase): class CenterNetConfigTest(tf.test.TestCase, parameterized.TestCase):
......
...@@ -18,9 +18,9 @@ from typing import Tuple ...@@ -18,9 +18,9 @@ from typing import Tuple
import tensorflow as tf import tensorflow as tf
from official.vision.beta.projects.centernet.ops import box_list from official.projects.centernet.ops import box_list
from official.vision.beta.projects.centernet.ops import box_list_ops from official.projects.centernet.ops import box_list_ops
from official.vision.beta.projects.centernet.ops import preprocess_ops as cn_prep_ops from official.projects.centernet.ops import preprocess_ops as cn_prep_ops
from official.vision.dataloaders import parser from official.vision.dataloaders import parser
from official.vision.dataloaders import utils from official.vision.dataloaders import utils
from official.vision.ops import box_ops from official.vision.ops import box_ops
......
...@@ -17,7 +17,7 @@ ...@@ -17,7 +17,7 @@
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf
from official.vision.beta.projects.centernet.losses import centernet_losses from official.projects.centernet.losses import centernet_losses
LOG_2 = np.log(2) LOG_2 = np.log(2)
LOG_3 = np.log(3) LOG_3 = np.log(3)
......
...@@ -19,7 +19,7 @@ from typing import Optional ...@@ -19,7 +19,7 @@ from typing import Optional
import tensorflow as tf import tensorflow as tf
from official.modeling import hyperparams from official.modeling import hyperparams
from official.vision.beta.projects.centernet.modeling.layers import cn_nn_blocks from official.projects.centernet.modeling.layers import cn_nn_blocks
from official.vision.modeling.backbones import factory from official.vision.modeling.backbones import factory
from official.vision.modeling.backbones import mobilenet from official.vision.modeling.backbones import mobilenet
from official.vision.modeling.layers import nn_blocks from official.vision.modeling.layers import nn_blocks
......
...@@ -18,9 +18,9 @@ from absl.testing import parameterized ...@@ -18,9 +18,9 @@ from absl.testing import parameterized
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf
from official.vision.beta.projects.centernet.common import registry_imports # pylint: disable=unused-import from official.projects.centernet.common import registry_imports # pylint: disable=unused-import
from official.vision.beta.projects.centernet.configs import backbones from official.projects.centernet.configs import backbones
from official.vision.beta.projects.centernet.modeling.backbones import hourglass from official.projects.centernet.modeling.backbones import hourglass
from official.vision.configs import common from official.vision.configs import common
......
...@@ -17,11 +17,11 @@ ...@@ -17,11 +17,11 @@
from absl.testing import parameterized from absl.testing import parameterized
import tensorflow as tf import tensorflow as tf
from official.vision.beta.projects.centernet.configs import backbones from official.projects.centernet.configs import backbones
from official.vision.beta.projects.centernet.modeling import centernet_model from official.projects.centernet.modeling import centernet_model
from official.vision.beta.projects.centernet.modeling.backbones import hourglass from official.projects.centernet.modeling.backbones import hourglass
from official.vision.beta.projects.centernet.modeling.heads import centernet_head from official.projects.centernet.modeling.heads import centernet_head
from official.vision.beta.projects.centernet.modeling.layers import detection_generator from official.projects.centernet.modeling.layers import detection_generator
from official.vision.configs import common from official.vision.configs import common
......
...@@ -14,11 +14,11 @@ ...@@ -14,11 +14,11 @@
"""Contains the definitions of head for CenterNet.""" """Contains the definitions of head for CenterNet."""
from typing import Any, Mapping, Dict, List from typing import Any, Dict, List, Mapping
import tensorflow as tf import tensorflow as tf
from official.vision.beta.projects.centernet.modeling.layers import cn_nn_blocks from official.projects.centernet.modeling.layers import cn_nn_blocks
class CenterNetHead(tf.keras.Model): class CenterNetHead(tf.keras.Model):
......
...@@ -18,7 +18,7 @@ from absl.testing import parameterized ...@@ -18,7 +18,7 @@ from absl.testing import parameterized
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf
from official.vision.beta.projects.centernet.modeling.heads import centernet_head from official.projects.centernet.modeling.heads import centernet_head
class CenterNetHeadTest(tf.test.TestCase, parameterized.TestCase): class CenterNetHeadTest(tf.test.TestCase, parameterized.TestCase):
......
...@@ -21,7 +21,7 @@ from absl.testing import parameterized ...@@ -21,7 +21,7 @@ from absl.testing import parameterized
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf
from official.vision.beta.projects.centernet.modeling.layers import cn_nn_blocks from official.projects.centernet.modeling.layers import cn_nn_blocks
from official.vision.modeling.layers import nn_blocks from official.vision.modeling.layers import nn_blocks
......
...@@ -25,8 +25,8 @@ from typing import Any, Mapping ...@@ -25,8 +25,8 @@ from typing import Any, Mapping
import tensorflow as tf import tensorflow as tf
from official.vision.beta.projects.centernet.ops import loss_ops from official.projects.centernet.ops import loss_ops
from official.vision.beta.projects.centernet.ops import nms_ops from official.projects.centernet.ops import nms_ops
from official.vision.ops import box_ops from official.vision.ops import box_ops
......
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