The goal of this project is to play around with ROS on Raspberry Pi 4 under Ubuntu 20.04.
- Use ROS to subscribe to the camera topic.
- Use CvBridge to convert the images from ROS to OpenCV.
- Use OpenCV to setup a YOLO_v4 pre-trained model to detect objects in the images.
- Process the output of YOLO_v4 object detector, and then draw bounding boxes on the image.
- Create a custom message to publish the information of bounding boxes to the ROS topic.
- Use CvBridge to convert the processed images(with bounding boxes on it) back to the ROS.
- Publish the processed images and the bounding boxes information.
- Publish video frames from Raspberry Pi camera v2 to ROS topic.
- Subscribe to the Raspberry Pi camera topic.
- Use Tensorflow Lite to convert the pre-trained YOLOv4-tiny model to
.tflite
file. - Use the TFLite model(YOLOv4-tiny, ssd-mobilenet) to do object detection in real time.
- Display the processed video frames(with bounding boxes on it).
- Raspberry Pi 4
- Ubuntu 20.04
- ROS Noetic
- OpenCV 3.4.12
- Python 3.8.5