"vscode:/vscode.git/clone" did not exist on "31b403526ec54abf13c4bb58dfb6635b4d2aa619"
Commit 824b70fb authored by Kaushik Shivakumar's avatar Kaushik Shivakumar
Browse files

region similarity calculator mod

parent 356c98bd
......@@ -112,9 +112,8 @@ class DETRSimilarity(RegionSimilarityCalculator):
"""
classification_scores = tf.matmul(groundtruth_labels,
tf.nn.softmax(predicted_labels), transpose_b=True)
return -5 * box_list_ops.l1(boxlist1, boxlist2) + \
2 * (1 - box_list_ops.giou_loss(boxlist1, boxlist2)) + \
classification_scores
return -5 * box_list_ops.l1(boxlist1, boxlist2) + 2 * box_list_ops.giou(
boxlist1, boxlist2) + classification_scores
class NegSqDistSimilarity(RegionSimilarityCalculator):
"""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