Skip to content

GitLab

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

Internal change 

PiperOrigin-RevId: 382462213
parent b1d973ea
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/vision/beta/serving/detection.py
View file @ 4f655acd
...@@ -17,10 +17,10 @@ ...@@ -17,10 +17,10 @@
import tensorflow as tf import tensorflow as tf
from cloud_tpu.models.detection.utils import box_utils
from official.vision.beta import configs from official.vision.beta import configs
from official.vision.beta.modeling import factory from official.vision.beta.modeling import factory
from official.vision.beta.ops import anchor from official.vision.beta.ops import anchor
from official.vision.beta.ops import box_ops
from official.vision.beta.ops import preprocess_ops from official.vision.beta.ops import preprocess_ops
from official.vision.beta.serving import export_base from official.vision.beta.serving import export_base
...@@ -140,7 +140,7 @@ class DetectionModule(export_base.ExportModule): ...@@ -140,7 +140,7 @@ class DetectionModule(export_base.ExportModule):
detection_boxes = ( detection_boxes = (
detections['detection_boxes'] / detections['detection_boxes'] /
tf.tile(image_info[:, 2:3, :], [1, 1, 2])) tf.tile(image_info[:, 2:3, :], [1, 1, 2]))
detections['detection_boxes'] = box_utils.normalize_boxes( detections['detection_boxes'] = box_ops.normalize_boxes(
detection_boxes, image_info[:, 0:1, :]) detection_boxes, image_info[:, 0:1, :])
# Cast num_detections and detection_classes to float. This allows the # Cast num_detections and detection_classes to float. This allows the
......
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