Skip to content

GitLab

"...resnet50_tensorflow.git" did not exist on "ee35a03041b642a2c1e4e5e528e48c0ab0a8afd6"
Commit 4f655acd authored by Abdullah Rashwan's avatar Abdullah Rashwan Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 382462213
parent b1d973ea
Hide 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 @@
import tensorflow as tf
from cloud_tpu.models.detection.utils import box_utils
from official.vision.beta import configs
from official.vision.beta.modeling import factory
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.serving import export_base
......@@ -140,7 +140,7 @@ class DetectionModule(export_base.ExportModule):
detection_boxes = (
detections['detection_boxes'] /
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, :])
# 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