Commit 9cd7a576 authored by Yeqing Li's avatar Yeqing Li Committed by A. Unique TensorFlower
Browse files

Internal change

PiperOrigin-RevId: 410121249
parent 4151365b
......@@ -24,9 +24,9 @@ from official.vision.beta.ops import anchor_generator
from official.vision.beta.ops import box_matcher
from official.vision.beta.ops import iou_similarity
from official.vision.beta.ops import target_gather
from official.vision.detection.utils.object_detection import balanced_positive_negative_sampler
from official.vision.detection.utils.object_detection import box_list
from official.vision.detection.utils.object_detection import faster_rcnn_box_coder
from official.vision.utils.object_detection import balanced_positive_negative_sampler
from official.vision.utils.object_detection import box_list
from official.vision.utils.object_detection import faster_rcnn_box_coder
class Anchor(object):
......
......@@ -23,11 +23,11 @@ import collections
import tensorflow as tf
from official.vision.beta.ops import iou_similarity
from official.vision.detection.utils import box_utils
from official.vision.detection.utils.object_detection import argmax_matcher
from official.vision.detection.utils.object_detection import balanced_positive_negative_sampler
from official.vision.detection.utils.object_detection import box_list
from official.vision.detection.utils.object_detection import faster_rcnn_box_coder
from official.vision.detection.utils.object_detection import target_assigner
from official.vision.utils.object_detection import argmax_matcher
from official.vision.utils.object_detection import balanced_positive_negative_sampler
from official.vision.utils.object_detection import box_list
from official.vision.utils.object_detection import faster_rcnn_box_coder
from official.vision.utils.object_detection import target_assigner
class Anchor(object):
......
......@@ -22,7 +22,7 @@ from absl import logging
import tensorflow as tf
from official.vision.detection.executor import distributed_executor as executor
from official.vision.detection.utils.object_detection import visualization_utils
from official.vision.utils.object_detection import visualization_utils
class DetectionDistributedExecutor(executor.DistributedExecutor):
......
......@@ -22,7 +22,7 @@ import tensorflow as tf
from official.vision.detection.ops import spatial_transform_ops
from official.vision.detection.utils import box_utils
from official.vision.detection.utils.object_detection import balanced_positive_negative_sampler
from official.vision.utils.object_detection import balanced_positive_negative_sampler
def box_matching(boxes, gt_boxes, gt_classes):
......
......@@ -19,7 +19,7 @@ import math
import tensorflow as tf
from official.vision.detection.utils import box_utils
from official.vision.detection.utils.object_detection import preprocessor
from official.vision.utils.object_detection import preprocessor
def pad_to_fixed_size(input_tensor, size, constant_values=0):
......
# Copyright 2021 The TensorFlow Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
......@@ -28,8 +28,8 @@ factory function for popular implementations.
"""
import tensorflow as tf
from official.vision.detection.utils.object_detection import matcher
from official.vision.detection.utils.object_detection import shape_utils
from official.vision.utils.object_detection import matcher
from official.vision.utils.object_detection import shape_utils
class ArgMaxMatcher(matcher.Matcher):
......
......@@ -33,8 +33,8 @@ This is originally implemented in TensorFlow Object Detection API.
import tensorflow as tf
from official.vision.detection.utils.object_detection import minibatch_sampler
from official.vision.detection.utils.object_detection import ops
from official.vision.utils.object_detection import minibatch_sampler
from official.vision.utils.object_detection import ops
class BalancedPositiveNegativeSampler(minibatch_sampler.MinibatchSampler):
......
......@@ -29,8 +29,8 @@ from __future__ import print_function
from six.moves import range
import tensorflow as tf
from official.vision.detection.utils.object_detection import box_list
from official.vision.detection.utils.object_detection import ops
from official.vision.utils.object_detection import box_list
from official.vision.utils.object_detection import ops
class SortOrder(object):
......
......@@ -29,8 +29,8 @@ Faster RCNN box coder follows the coding schema described below:
import tensorflow as tf
from official.vision.detection.utils.object_detection import box_coder
from official.vision.detection.utils.object_detection import box_list
from official.vision.utils.object_detection import box_coder
from official.vision.utils.object_detection import box_list
EPSILON = 1e-8
......
......@@ -34,7 +34,7 @@ from abc import abstractmethod
import tensorflow as tf
from official.vision.detection.utils.object_detection import ops
from official.vision.utils.object_detection import ops
class MinibatchSampler(object):
......
......@@ -19,7 +19,7 @@ This is originally implemented in TensorFlow Object Detection API.
import tensorflow as tf
from official.vision.detection.utils.object_detection import shape_utils
from official.vision.utils.object_detection import shape_utils
def indices_to_dense_vector(indices,
......
......@@ -39,11 +39,9 @@ we pass it to the functions. At the end of the preprocess we expand the image
back to rank 4.
"""
import tensorflow as tf
import numpy as np
from official.vision.detection.utils.object_detection import box_list
import tensorflow as tf
from official.vision.utils.object_detection import box_list
def _flip_boxes_left_right(boxes):
......
......@@ -33,8 +33,8 @@ images must be handled externally.
import tensorflow as tf
from official.vision.detection.utils.object_detection import box_list
from official.vision.detection.utils.object_detection import shape_utils
from official.vision.utils.object_detection import box_list
from official.vision.utils.object_detection import shape_utils
KEYPOINTS_FIELD_NAME = 'keypoints'
......
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