Ignoring markers which are too far from the center. to avoid optical distorsion.

2 files changed, 15 insertions, 0 deletions

diff --git a/src/argaze/ArUcoMarkers/ArUcoTracker.py b/src/argaze/ArUcoMarkers/ArUcoTracker.py
index 3ce5e2e..57214c1 100644
--- a/src/argaze/ArUcoMarkers/ArUcoTracker.py
+++ b/src/argaze/ArUcoMarkers/ArUcoTracker.py
@@ -201,6 +201,14 @@ class ArUcoTracker():
         """Get marker i corners."""
         return self.__markers_corners[i]
 
+    def get_marker_center(self, i):
+        """Get marker i center coordinates."""
+        
+        sum_x = numpy.sum(self.__markers_corners[i][:, 0])
+        sum_y = numpy.sum(self.__markers_corners[i][:, 1])
+
+        return sum_x/4, sum_y/4
+
     def get_marker_rotation(self, i):
         """Get marker i rotation vector."""
         return self.__rvecs[i]
diff --git a/src/argaze/utils/export_tobii_segment_aruco_visual_scan.py b/src/argaze/utils/export_tobii_segment_aruco_visual_scan.py
index 9ca836b..4174949 100644
--- a/src/argaze/utils/export_tobii_segment_aruco_visual_scan.py
+++ b/src/argaze/utils/export_tobii_segment_aruco_visual_scan.py
@@ -219,6 +219,13 @@ def main():
                         if aoi3D_scene == None:
                             continue
 
+                        # Ignore marker too far from center
+                        marker_x, marker_y = aruco_tracker.get_marker_center(i)
+                        distance_to_center = ( (video_frame.width/2 - marker_x)**2 + (video_frame.height/2 - marker_y)**2 )**0.5
+                        
+                        if distance_to_center > 900:
+                            continue
+
                         aoi3D_scene.rotation = aruco_tracker.get_marker_rotation(i)
                         aoi3D_scene.translation = aruco_tracker.get_marker_translation(i)