Commit 22b5b0c6 authored by Kaushik Shivakumar's avatar Kaushik Shivakumar
Browse files

self code-review to clean up

parent 4022aae5
......@@ -225,7 +225,7 @@ class WeightedGIOULocalizationLoss(Loss):
Args:
prediction_tensor: A float tensor of shape [batch_size, num_anchors, 4]
representing the predicted boxes in the form
representing the decoded predicted boxes
target_tensor: A float tensor of shape [batch_size, num_anchors, 4]
representing the decoded target boxes
weights: a float tensor of shape [batch_size, num_anchors]
......
......@@ -28,7 +28,6 @@ from object_detection.core import box_list
from object_detection.core import losses
from object_detection.core import matcher
from object_detection.utils import test_case
from object_detection.utils import ops
class WeightedL2LocalizationLossTest(test_case.TestCase):
......
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