1

2025-11-28 16:38:23 +08:00
parent 016e1d2a67
commit 5ea4a56561
1 changed files with 1 additions and 1 deletions
--- a/vision_detect/vision_detect/VisionDetect/node/detect_node.py
+++ b/vision_detect/vision_detect/VisionDetect/node/detect_node.py
@@ -207,7 +207,7 @@ class DetectNode(Node):
            raise 
        self.get_logger().info(f'Loading checkpoint from: {self.checkpoint_path}')

-        if self.checkpoint_path.endswith('-seg.pt'):
+        if self.checkpoint_path.endswith('-seg.pt') or self.checkpoint_path.endswith('.onnx') or self.checkpoint_path.endswith('.engine'):
            self.function = self._seg_image
        else:
            self.function = None