Merge pull request #183 from ArcturusNavigation/yolov8_shape_training

Yolov8 shape training
ArcturusNavigation · Feb 7, 2025 · 2faaed9 · 2faaed9
2 parents 73dc314 + 4b82d9e
commit 2faaed9
Show file tree

Hide file tree

Showing 4 changed files with 520 additions and 1 deletion.
diff --git a/all_seaing_perception/all_seaing_perception/yolov8_node.py b/all_seaing_perception/all_seaing_perception/yolov8_node.py
@@ -79,7 +79,6 @@ def __init__(self) -> None:
             self.model = YOLO(pt_path)
         else:
             self.get_logger().error(f"Both model paths do not exist :( TensorRT: {engine_path} and pt: {pt_path}")
-
         # Setup QoS profile
         image_qos_profile = QoSProfile(
             reliability=QoSReliabilityPolicy.BEST_EFFORT,

diff --git a/.../all_seaing_perception/yolov8_train.ipynb → ...eaing_perception/yolov8_train_buoys.ipynb b/.../all_seaing_perception/yolov8_train.ipynb → ...eaing_perception/yolov8_train_buoys.ipynb