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 @@
"""All necessary imports for registration."""
# 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.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
from official.core import exp_factory
from official.modeling import hyperparams
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
......
......@@ -19,8 +19,8 @@ import tensorflow as tf
from official.core import config_definitions as cfg
from official.core import exp_factory
from official.vision.beta.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.common import registry_imports # pylint: disable=unused-import
from official.projects.centernet.configs import centernet as exp_cfg
class CenterNetConfigTest(tf.test.TestCase, parameterized.TestCase):
......
......@@ -18,9 +18,9 @@ from typing import Tuple
import tensorflow as tf
from official.vision.beta.projects.centernet.ops import box_list
from official.vision.beta.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 box_list
from official.projects.centernet.ops import box_list_ops
from official.projects.centernet.ops import preprocess_ops as cn_prep_ops
from official.vision.dataloaders import parser
from official.vision.dataloaders import utils
from official.vision.ops import box_ops
......
......@@ -17,7 +17,7 @@
import numpy as np
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_3 = np.log(3)
......
......@@ -19,7 +19,7 @@ from typing import Optional
import tensorflow as tf
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 mobilenet
from official.vision.modeling.layers import nn_blocks
......
......@@ -18,9 +18,9 @@ from absl.testing import parameterized
import numpy as np
import tensorflow as tf
from official.vision.beta.projects.centernet.common import registry_imports # pylint: disable=unused-import
from official.vision.beta.projects.centernet.configs import backbones
from official.vision.beta.projects.centernet.modeling.backbones import hourglass
from official.projects.centernet.common import registry_imports # pylint: disable=unused-import
from official.projects.centernet.configs import backbones
from official.projects.centernet.modeling.backbones import hourglass
from official.vision.configs import common
......
......@@ -17,11 +17,11 @@
from absl.testing import parameterized
import tensorflow as tf
from official.vision.beta.projects.centernet.configs import backbones
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.modeling.heads import centernet_head
from official.vision.beta.projects.centernet.modeling.layers import detection_generator
from official.projects.centernet.configs import backbones
from official.projects.centernet.modeling import centernet_model
from official.projects.centernet.modeling.backbones import hourglass
from official.projects.centernet.modeling.heads import centernet_head
from official.projects.centernet.modeling.layers import detection_generator
from official.vision.configs import common
......
......@@ -14,11 +14,11 @@
"""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
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):
......
......@@ -18,7 +18,7 @@ from absl.testing import parameterized
import numpy as np
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):
......
......@@ -21,7 +21,7 @@ from absl.testing import parameterized
import numpy as np
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
......
......@@ -25,8 +25,8 @@ from typing import Any, Mapping
import tensorflow as tf
from official.vision.beta.projects.centernet.ops import loss_ops
from official.vision.beta.projects.centernet.ops import nms_ops
from official.projects.centernet.ops import loss_ops
from official.projects.centernet.ops import nms_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