Internal change

PiperOrigin-RevId: 397402432

official/vision/beta/ops/spatial_transform_ops.py

@@ -200,9 +200,10 @@ def multilevel_crop_and_resize(features,
     box_height = boxes[:, :, 2] - boxes[:, :, 0]
     areas_sqrt = tf.sqrt(
         tf.cast(box_height, tf.float32) * tf.cast(box_width, tf.float32))
+
     levels = tf.cast(
         tf.math.floordiv(
-            tf.math.log(tf.divide(areas_sqrt, 224.0)),
+            tf.math.log(tf.math.divide_no_nan(areas_sqrt, 224.0)),
             tf.math.log(2.0)) + 4.0,
         dtype=tf.int32)
     # Maps levels between [min_level, max_level].