lowered threshold for positive _is_equivalent function

k4ntz · Nov 27, 2024 · 6b46eaa · 6b46eaa
1 parent 3cc0502
commit 6b46eaa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ocatari/ram/game_objects.py b/ocatari/ram/game_objects.py
@@ -252,7 +252,7 @@ def _is_equivalent(self, other):
         if self.category != other.category:
             return False
         iou_value = self.iou(other)
-        return iou_value > 0.9
+        return iou_value > 0.8
 
     def iou(self, other):
         # Calculate the (x, y) coordinates of the intersection rectangle

diff --git a/ocatari/vision/game_objects.py b/ocatari/vision/game_objects.py
@@ -204,7 +204,7 @@ def _is_equivalent(self, other):
         if self.category != other.category:
             return False
         iou_value = self.iou(other)
-        return iou_value > 0.9
+        return iou_value > 0.8
 
     def iou(self, other):
         # Calculate the (x, y) coordinates of the intersection rectangle