merged

official/vision/beta/projects/yolo/ops/box_ops_test.py

@@ -49,6 +49,5 @@ class InputUtilsTest(parameterized.TestCase, tf.test.TestCase):
     self.assertArrayNear(ciou, expected_iou, 0.001)
     self.assertArrayNear(diou, expected_iou, 0.001)
 
-
 if __name__ == '__main__':
-  tf.test.main()
+  tf.test.main()
\ No newline at end of file

official/vision/beta/projects/yolo/ops/math_ops.py

@@ -3,7 +3,7 @@ import tensorflow as tf
 
 
 def rm_nan_inf(x, val=0.0):
-  """remove nan and infinity
+  """Remove nan and infinity
 
   Args:
     x: any `Tensor` of any type.

official/vision/beta/projects/yolo/ops/nms_ops.py

@@ -8,7 +8,7 @@ class TiledNMS():
   IOU_TYPES = {'diou': 0, 'giou': 1, 'ciou': 2, 'iou': 3}
 
   def __init__(self, iou_type='diou', beta=0.6):
-    '''initialization for all non max suppression operations mainly used to
+    '''Initialization for all non max suppression operations mainly used to
     select hyperparameters for the iou type and scaling.
 
     Args: