(pose_estimation) more stricter regex

tue-robotics · Feb 14, 2024 · 76eddd9 · 76eddd9
1 parent b3bd26f
commit 76eddd9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/image_recognition_pose_estimation/src/image_recognition_pose_estimation/yolo_pose_wrapper.py b/image_recognition_pose_estimation/src/image_recognition_pose_estimation/yolo_pose_wrapper.py
@@ -10,7 +10,8 @@
 from ultralytics import YOLO
 from ultralytics.engine.results import Results
 
-YOLO_POSE_PATTERN = re.compile(r"^yolov8[nsmlx]-pose(-p6)?.pt$")
+
+YOLO_POSE_PATTERN = re.compile(r"^yolov8(?:([nsml])|(x))-pose(?(2)-p6|)?.pt$")
 
 
 class YoloPoseWrapper: