Skip to content

GitLab

Commit b807ff86 authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

tensorflow.compat.v2 is no longer needed. 

PiperOrigin-RevId: 310289012
parent e2a41868
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/vision/detection/utils/object_detection/target_assigner.py
View file @ b807ff86
...@@ -31,7 +31,7 @@ Note that TargetAssigners only operate on detections from a single ...@@ -31,7 +31,7 @@ Note that TargetAssigners only operate on detections from a single
image at a time, so any logic for applying a TargetAssigner to multiple image at a time, so any logic for applying a TargetAssigner to multiple
images must be handled externally. images must be handled externally.
""" """
import tensorflow.compat.v2 as tf import tensorflow as tf
from official.vision.detection.utils.object_detection import box_list from official.vision.detection.utils.object_detection import box_list
from official.vision.detection.utils.object_detection import shape_utils from official.vision.detection.utils.object_detection import shape_utils
......
official/vision/detection/utils/object_detection/visualization_utils.py
View file @ b807ff86
...@@ -31,7 +31,7 @@ import PIL.ImageColor as ImageColor ...@@ -31,7 +31,7 @@ import PIL.ImageColor as ImageColor
import PIL.ImageDraw as ImageDraw import PIL.ImageDraw as ImageDraw
import PIL.ImageFont as ImageFont import PIL.ImageFont as ImageFont
import six import six
import tensorflow.compat.v2 as tf import tensorflow as tf
from official.vision.detection.utils import box_utils from official.vision.detection.utils import box_utils
from official.vision.detection.utils.object_detection import shape_utils from official.vision.detection.utils.object_detection import shape_utils
......
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