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

Internal change

PiperOrigin-RevId: 473326146
parent c0f52079
...@@ -16,7 +16,7 @@ ...@@ -16,7 +16,7 @@
from absl.testing import parameterized from absl.testing import parameterized
import tensorflow as tf import tensorflow as tf
from official.vision.beta.projects.yolo.modeling.layers import detection_generator as dg from official.projects.yolo.modeling.layers import detection_generator as dg
class YoloDecoderTest(parameterized.TestCase, tf.test.TestCase): class YoloDecoderTest(parameterized.TestCase, tf.test.TestCase):
......
...@@ -16,7 +16,7 @@ from absl.testing import parameterized ...@@ -16,7 +16,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.yolo.modeling.layers import nn_blocks from official.projects.yolo.modeling.layers import nn_blocks
class CSPConnectTest(tf.test.TestCase, parameterized.TestCase): class CSPConnectTest(tf.test.TestCase, parameterized.TestCase):
......
...@@ -16,7 +16,7 @@ ...@@ -16,7 +16,7 @@
from typing import Mapping, Union from typing import Mapping, Union
import tensorflow as tf import tensorflow as tf
from official.vision.beta.projects.yolo.modeling.layers import nn_blocks from official.projects.yolo.modeling.layers import nn_blocks
class Yolo(tf.keras.Model): class Yolo(tf.keras.Model):
......
...@@ -16,9 +16,9 @@ ...@@ -16,9 +16,9 @@
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf
from official.vision.beta.projects.yolo.ops import box_ops from official.projects.yolo.ops import box_ops
from official.vision.beta.projects.yolo.ops import loss_utils from official.projects.yolo.ops import loss_utils
from official.vision.beta.projects.yolo.ops import preprocessing_ops from official.projects.yolo.ops import preprocessing_ops
INF = 10000000 INF = 10000000
......
...@@ -15,7 +15,7 @@ ...@@ -15,7 +15,7 @@
"""Yolo box ops.""" """Yolo box ops."""
import math import math
import tensorflow as tf import tensorflow as tf
from official.vision.beta.projects.yolo.ops import math_ops from official.projects.yolo.ops import math_ops
def yxyx_to_xcycwh(box: tf.Tensor): def yxyx_to_xcycwh(box: tf.Tensor):
......
...@@ -17,7 +17,7 @@ from absl.testing import parameterized ...@@ -17,7 +17,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.yolo.ops import box_ops from official.projects.yolo.ops import box_ops
class InputUtilsTest(parameterized.TestCase, tf.test.TestCase): class InputUtilsTest(parameterized.TestCase, tf.test.TestCase):
......
...@@ -19,7 +19,7 @@ import numpy as np ...@@ -19,7 +19,7 @@ import numpy as np
import tensorflow as tf import tensorflow as tf
from official.core import input_reader from official.core import input_reader
from official.vision.beta.projects.yolo.ops import box_ops from official.projects.yolo.ops import box_ops
def _iou(x, centroids_x, iou_type="iou"): def _iou(x, centroids_x, iou_type="iou"):
......
...@@ -17,7 +17,7 @@ from absl.testing import parameterized ...@@ -17,7 +17,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.yolo.ops import kmeans_anchors from official.projects.yolo.ops import kmeans_anchors
class KMeansTest(parameterized.TestCase, tf.test.TestCase): class KMeansTest(parameterized.TestCase, tf.test.TestCase):
......
...@@ -17,8 +17,8 @@ ...@@ -17,8 +17,8 @@
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf
from official.vision.beta.projects.yolo.ops import box_ops from official.projects.yolo.ops import box_ops
from official.vision.beta.projects.yolo.ops import math_ops from official.projects.yolo.ops import math_ops
@tf.custom_gradient @tf.custom_gradient
......
...@@ -18,7 +18,7 @@ import random ...@@ -18,7 +18,7 @@ import random
import tensorflow as tf import tensorflow as tf
import tensorflow_addons as tfa import tensorflow_addons as tfa
from official.vision.beta.projects.yolo.ops import preprocessing_ops from official.projects.yolo.ops import preprocessing_ops
from official.vision.ops import box_ops from official.vision.ops import box_ops
from official.vision.ops import preprocess_ops from official.vision.ops import preprocess_ops
......
...@@ -17,7 +17,7 @@ from absl.testing import parameterized ...@@ -17,7 +17,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.yolo.ops import preprocessing_ops from official.projects.yolo.ops import preprocessing_ops
from official.vision.ops import box_ops as bbox_ops from official.vision.ops import box_ops as bbox_ops
......
...@@ -17,6 +17,6 @@ ...@@ -17,6 +17,6 @@
# pylint: disable=wildcard-import # pylint: disable=wildcard-import
from official.modeling.optimization.configs.learning_rate_config import * from official.modeling.optimization.configs.learning_rate_config import *
from official.modeling.optimization.ema_optimizer import ExponentialMovingAverage from official.modeling.optimization.ema_optimizer import ExponentialMovingAverage
from official.vision.beta.projects.yolo.optimization.configs.optimization_config import * from official.projects.yolo.optimization.configs.optimization_config import *
from official.vision.beta.projects.yolo.optimization.configs.optimizer_config import * from official.projects.yolo.optimization.configs.optimizer_config import *
from official.vision.beta.projects.yolo.optimization.optimizer_factory import OptimizerFactory as YoloOptimizerFactory from official.projects.yolo.optimization.optimizer_factory import OptimizerFactory as YoloOptimizerFactory
...@@ -22,7 +22,7 @@ import dataclasses ...@@ -22,7 +22,7 @@ import dataclasses
from typing import Optional from typing import Optional
from official.modeling.optimization.configs import optimization_config as optimization_cfg from official.modeling.optimization.configs import optimization_config as optimization_cfg
from official.vision.beta.projects.yolo.optimization.configs import optimizer_config as opt_cfg from official.projects.yolo.optimization.configs import optimizer_config as opt_cfg
@dataclasses.dataclass @dataclasses.dataclass
......
...@@ -18,7 +18,7 @@ import gin ...@@ -18,7 +18,7 @@ import gin
from official.modeling.optimization import ema_optimizer from official.modeling.optimization import ema_optimizer
from official.modeling.optimization import optimizer_factory from official.modeling.optimization import optimizer_factory
from official.vision.beta.projects.yolo.optimization import sgd_torch from official.projects.yolo.optimization import sgd_torch
optimizer_factory.OPTIMIZERS_CLS.update({ optimizer_factory.OPTIMIZERS_CLS.update({
'sgd_torch': sgd_torch.SGDTorch, 'sgd_torch': sgd_torch.SGDTorch,
......
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