"git@developer.sourcefind.cn:wangsen/mineru.git" did not exist on "bc46253a1736796c6a42a75e8c8e63f8162a41b1"
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): ...@@ -225,7 +225,7 @@ class WeightedGIOULocalizationLoss(Loss):
Args: Args:
prediction_tensor: A float tensor of shape [batch_size, num_anchors, 4] 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] target_tensor: A float tensor of shape [batch_size, num_anchors, 4]
representing the decoded target boxes representing the decoded target boxes
weights: a float tensor of shape [batch_size, num_anchors] weights: a float tensor of shape [batch_size, num_anchors]
......
...@@ -28,7 +28,6 @@ from object_detection.core import box_list ...@@ -28,7 +28,6 @@ from object_detection.core import box_list
from object_detection.core import losses from object_detection.core import losses
from object_detection.core import matcher from object_detection.core import matcher
from object_detection.utils import test_case from object_detection.utils import test_case
from object_detection.utils import ops
class WeightedL2LocalizationLossTest(test_case.TestCase): 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