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 @@
from absl.testing import parameterized
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):
......
......@@ -16,7 +16,7 @@ from absl.testing import parameterized
import numpy as np
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):
......
......@@ -16,7 +16,7 @@
from typing import Mapping, Union
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):
......
......@@ -16,9 +16,9 @@
import numpy as np
import tensorflow as tf
from official.vision.beta.projects.yolo.ops import box_ops
from official.vision.beta.projects.yolo.ops import loss_utils
from official.vision.beta.projects.yolo.ops import preprocessing_ops
from official.projects.yolo.ops import box_ops
from official.projects.yolo.ops import loss_utils
from official.projects.yolo.ops import preprocessing_ops
INF = 10000000
......
......@@ -15,7 +15,7 @@
"""Yolo box ops."""
import math
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):
......
......@@ -17,7 +17,7 @@ from absl.testing import parameterized
import numpy as np
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):
......
......@@ -19,7 +19,7 @@ import numpy as np
import tensorflow as tf
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"):
......
......@@ -17,7 +17,7 @@ from absl.testing import parameterized
import numpy as np
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):
......
......@@ -17,8 +17,8 @@
import numpy as np
import tensorflow as tf
from official.vision.beta.projects.yolo.ops import box_ops
from official.vision.beta.projects.yolo.ops import math_ops
from official.projects.yolo.ops import box_ops
from official.projects.yolo.ops import math_ops
@tf.custom_gradient
......
......@@ -18,7 +18,7 @@ import random
import tensorflow as tf
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 preprocess_ops
......
......@@ -17,7 +17,7 @@ from absl.testing import parameterized
import numpy as np
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
......
......@@ -17,6 +17,6 @@
# pylint: disable=wildcard-import
from official.modeling.optimization.configs.learning_rate_config import *
from official.modeling.optimization.ema_optimizer import ExponentialMovingAverage
from official.vision.beta.projects.yolo.optimization.configs.optimization_config import *
from official.vision.beta.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.configs.optimization_config import *
from official.projects.yolo.optimization.configs.optimizer_config import *
from official.projects.yolo.optimization.optimizer_factory import OptimizerFactory as YoloOptimizerFactory
......@@ -22,7 +22,7 @@ import dataclasses
from typing import Optional
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
......
......@@ -18,7 +18,7 @@ import gin
from official.modeling.optimization import ema_optimizer
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({
'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