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

Internal change

PiperOrigin-RevId: 473326146
parent c0f52079
...@@ -15,8 +15,8 @@ ...@@ -15,8 +15,8 @@
"""Detection Data parser and processing for YOLO.""" """Detection Data parser and processing for YOLO."""
import tensorflow as tf import tensorflow as tf
from official.vision.beta.projects.yolo.ops import anchor from official.projects.yolo.ops import anchor
from official.vision.beta.projects.yolo.ops import preprocessing_ops from official.projects.yolo.ops import preprocessing_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 as bbox_ops from official.vision.ops import box_ops as bbox_ops
......
...@@ -19,9 +19,9 @@ import functools ...@@ -19,9 +19,9 @@ import functools
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 math_ops from official.projects.yolo.ops import math_ops
class YoloLossBase(object, metaclass=abc.ABCMeta): class YoloLossBase(object, metaclass=abc.ABCMeta):
......
...@@ -17,7 +17,7 @@ ...@@ -17,7 +17,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.losses import yolo_loss from official.projects.yolo.losses import yolo_loss
class YoloDecoderTest(parameterized.TestCase, tf.test.TestCase): class YoloDecoderTest(parameterized.TestCase, tf.test.TestCase):
......
...@@ -40,7 +40,7 @@ import collections ...@@ -40,7 +40,7 @@ import collections
import tensorflow as tf import tensorflow as tf
from official.modeling import hyperparams from official.modeling import hyperparams
from official.vision.beta.projects.yolo.modeling.layers import nn_blocks from official.projects.yolo.modeling.layers import nn_blocks
from official.vision.modeling.backbones import factory from official.vision.modeling.backbones import factory
......
...@@ -20,7 +20,7 @@ import tensorflow as tf ...@@ -20,7 +20,7 @@ import tensorflow as tf
from tensorflow.python.distribute import combinations from tensorflow.python.distribute import combinations
from tensorflow.python.distribute import strategy_combinations from tensorflow.python.distribute import strategy_combinations
from official.vision.beta.projects.yolo.modeling.backbones import darknet from official.projects.yolo.modeling.backbones import darknet
class DarknetTest(parameterized.TestCase, tf.test.TestCase): class DarknetTest(parameterized.TestCase, tf.test.TestCase):
......
...@@ -18,7 +18,7 @@ from typing import Mapping, Optional, Union ...@@ -18,7 +18,7 @@ from typing import Mapping, Optional, Union
import tensorflow as tf import tensorflow as tf
from official.modeling import hyperparams from official.modeling import hyperparams
from official.vision.beta.projects.yolo.modeling.layers import nn_blocks from official.projects.yolo.modeling.layers import nn_blocks
from official.vision.modeling.decoders import factory from official.vision.modeling.decoders import factory
# model configurations # model configurations
......
...@@ -20,7 +20,7 @@ import tensorflow as tf ...@@ -20,7 +20,7 @@ import tensorflow as tf
from tensorflow.python.distribute import combinations from tensorflow.python.distribute import combinations
from tensorflow.python.distribute import strategy_combinations from tensorflow.python.distribute import strategy_combinations
from official.vision.beta.projects.yolo.modeling.decoders import yolo_decoder as decoders from official.projects.yolo.modeling.decoders import yolo_decoder as decoders
class YoloDecoderTest(parameterized.TestCase, tf.test.TestCase): class YoloDecoderTest(parameterized.TestCase, tf.test.TestCase):
......
...@@ -16,10 +16,10 @@ ...@@ -16,10 +16,10 @@
from absl import logging from absl import logging
from official.vision.beta.projects.yolo.configs import yolo from official.projects.yolo.configs import yolo
from official.vision.beta.projects.yolo.modeling import yolo_model from official.projects.yolo.modeling import yolo_model
from official.vision.beta.projects.yolo.modeling.heads import yolo_head from official.projects.yolo.modeling.heads import yolo_head
from official.vision.beta.projects.yolo.modeling.layers import detection_generator from official.projects.yolo.modeling.layers import detection_generator
from official.vision.modeling.backbones import factory as backbone_factory from official.vision.modeling.backbones import factory as backbone_factory
from official.vision.modeling.decoders import factory as decoder_factory from official.vision.modeling.decoders import factory as decoder_factory
......
...@@ -15,7 +15,7 @@ ...@@ -15,7 +15,7 @@
"""Yolo heads.""" """Yolo heads."""
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 YoloHead(tf.keras.layers.Layer): class YoloHead(tf.keras.layers.Layer):
......
...@@ -18,7 +18,7 @@ ...@@ -18,7 +18,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.heads import yolo_head as heads from official.projects.yolo.modeling.heads import yolo_head as heads
class YoloDecoderTest(parameterized.TestCase, tf.test.TestCase): class YoloDecoderTest(parameterized.TestCase, tf.test.TestCase):
......
...@@ -15,9 +15,9 @@ ...@@ -15,9 +15,9 @@
"""Contains common building blocks for yolo layer (detection layer).""" """Contains common building blocks for yolo layer (detection layer)."""
import tensorflow as tf import tensorflow as tf
from official.vision.beta.projects.yolo.losses import yolo_loss from official.projects.yolo.losses import yolo_loss
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.modeling.layers import detection_generator from official.vision.modeling.layers import detection_generator
......
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