This project contains a Gazebo simulation with all the LOCO dataset and the chance to perform object detection using YOLOv8 n with this data.
- Clone repository
cd ros2_ws/src
git clone https://github.com/vidaldani/Logistic-object-detection.git
- Compile workspace
cd ..
colcon build
source install/setup.bash
- Export Turtlebot model
gedit ~/.bashrc
# copy this instruction at the end: export TURTLEBOT3_MODEL=waffle
source ~/.bashrc
- Copy the models inside /turtlebot3_gazebo/models/logistic_objects to ~/.gazebo/models folder
cp -r ~/ros2_ws/src/Logistic-object-detection/src/turtlebot3_gazebo/models/logistic_objects/* ~/.gazebo/models/
- Launch logistic simulation:
ros2 launch turtlebot3_gazebo versuchshalle.launch.py
- Install ultralytics (if it is already installed jump to step 2)
pip3 install ultralytics==8.3.40
- On a different terminal launch logistic object detection
cd ros2_ws/ source install/setup.bash ros2 launch turtlebot3_recognition launch_yolov8.launch.py
- Open rviz to visualize the inference results. The configuration file can be found in the folder /turtlebot3_recognition/rviz
rviz2 -d $(pwd)/src/turtlebot3_ws/src/turtlebot3_recognition/rviz/config.rviz
- In a new terminal launch the keyboard teleoperation to move the robot around and perform object detection
ros2 run teleop_twist_keyboard teleop_twist_keyboard
- Launch 3D object detection
ros2 launch turtlebot3_recognition launch_3d_node.launch.py