This repository contains a ROS node that utilizes the PointPillar neural network model (MMDetection3d) to process point cloud data from ROSbags. The repository is specifically designed to work in a ROS 1 environment.
Please note that the instructions provided assume a ROS Noetic environment. The repo is tested with Ubuntu 20.
Before using this repository, please ensure that you have the following prerequisites installed:
- CUDA 11.6 and MMDetection3D library:
- You can follow the [detailed installation instruction] to install both.
- ROS Noetic
jsk-recognition-msgs
,jsk-rviz-plugins
:sudo apt-get install ros-noetic-jsk-recognition-msgs ros-noetic-jsk-rviz-plugins
Additionally, you will need to download the required models weight and config files from model zoo and place them in directory: pointpillar_ros/model/pointpillar/
. Update model weight and confi files' names in launchfile.
To use this code, follow the steps below:
- Clone this repository to your
catkin_ws/src
directory. - Update the point cloud topic and model names in launchfile.
- Build the package:
catkin config --install catkin build pointpillar_ros
- Launch the ROS node by executing the following commands:
source install/setup.bash # or source install/pointpillar_ros/setup.bash roslaunch pointpillar_ros pointpillar_ros.launch
- Launch RViz, visualize the detection bounding boxes.
This is a screenshot of detection result:
If you encounter a ModuleNotFoundError
related to the netifaces
module, you can resolve it by installing the module using pip pip install netifaces
.
Hope this helps! If you have any further questions, feel free to ask.