Skip to content

GitLab

"...resnet50_tensorflow.git" did not exist on "3d97b007cdb2c73f23daf9141437796095436c22"
Commit 824b70fb authored by Kaushik Shivakumar's avatar Kaushik Shivakumar
Browse files

region similarity calculator mod

parent 356c98bd
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
research/object_detection/core/region_similarity_calculator.py
View file @ 824b70fb
...@@ -112,9 +112,8 @@ class DETRSimilarity(RegionSimilarityCalculator): ...@@ -112,9 +112,8 @@ class DETRSimilarity(RegionSimilarityCalculator):
""" """
classification_scores = tf.matmul(groundtruth_labels, classification_scores = tf.matmul(groundtruth_labels,
tf.nn.softmax(predicted_labels), transpose_b=True) tf.nn.softmax(predicted_labels), transpose_b=True)
return -5 * box_list_ops.l1(boxlist1, boxlist2) + \ return -5 * box_list_ops.l1(boxlist1, boxlist2) + 2 * box_list_ops.giou(
2 * (1 - box_list_ops.giou_loss(boxlist1, boxlist2)) + \ boxlist1, boxlist2) + classification_scores
classification_scores
class NegSqDistSimilarity(RegionSimilarityCalculator): class NegSqDistSimilarity(RegionSimilarityCalculator):
"""Class to compute similarity based on the squared distance metric. """Class to compute similarity based on the squared distance metric.
......
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