try/except for file not exists and fuse

mgonzs13 · Nov 13, 2024 · d587d43 · d587d43
1 parent b7a5d6f
commit d587d43
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/yolo_ros/yolo_ros/yolo_node.py b/yolo_ros/yolo_ros/yolo_node.py
@@ -127,8 +127,17 @@ def on_configure(self, state: LifecycleState) -> TransitionCallbackReturn:
     def on_activate(self, state: LifecycleState) -> TransitionCallbackReturn:
         self.get_logger().info(f"[{self.get_name()}] Activating...")
 
-        self.yolo = self.type_to_model[self.model_type](self.model)
-        self.yolo.fuse()
+        try:
+            self.yolo = self.type_to_model[self.model_type](self.model)
+        except FileNotFoundError:
+            self.get_logger().error(f"Model file '{self.model}' does not exists")
+            return TransitionCallbackReturn.ERROR
+
+        try:
+            self.get_logger().info("Trying to fuse model...")
+            self.yolo.fuse()
+        except TypeError as e:
+            self.get_logger().warn(f"Error while fuse: {e}")
 
         self._enable_srv = self.create_service(SetBool, "enable", self.enable_cb)