From 6b46eaa7bea73bb5d36d2825724051393e8fdcd8 Mon Sep 17 00:00:00 2001 From: BJenrajb Date: Wed, 27 Nov 2024 17:10:36 +0100 Subject: [PATCH] lowered threshold for positive _is_equivalent function --- ocatari/ram/game_objects.py | 2 +- ocatari/vision/game_objects.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/ocatari/ram/game_objects.py b/ocatari/ram/game_objects.py index 40c9c1d3..0285a859 100644 --- 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 index 4ff0ab05..9ad805de 100644 --- 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