Internal change

PiperOrigin-RevId: 515247220

Author: fyangf

official/vision/modeling/layers/detection_generator.py

@@ -661,9 +661,11 @@ def __call__(self,
     decoded_boxes = box_ops.decode_boxes(
         raw_boxes, anchor_boxes, weights=regression_weights)
 
-    # Box clipping
-    decoded_boxes = box_ops.clip_boxes(
-        decoded_boxes, tf.expand_dims(image_shape, axis=1))
+    # Box clipping.
+    if image_shape is not None:
+      decoded_boxes = box_ops.clip_boxes(
+          decoded_boxes, tf.expand_dims(image_shape, axis=1)
+      )
 
     if bbox_per_class:
       decoded_boxes = tf.reshape(
@@ -835,8 +837,10 @@ def _decode_multilevel_outputs(
       boxes_i = box_ops.decode_boxes(raw_boxes_i, anchor_boxes_i)
 
       # Box clipping.
-      boxes_i = box_ops.clip_boxes(
-          boxes_i, tf.expand_dims(image_shape, axis=1))
+      if image_shape is not None:
+        boxes_i = box_ops.clip_boxes(
+            boxes_i, tf.expand_dims(image_shape, axis=1)
+        )
 
       boxes.append(boxes_i)
       scores.append(scores_i)