3
3

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?

ROS2とOpenCVによる画像処理入門1

Last updated at Posted at 2024-10-23

はじめに

ロボット開発のチームでは,自動運転時の画像処理や物体認識に関わる部分の開発を任されることになりました.今回はチュートリアルとして,C++およびopenCVを用いてランダム画像の生成と,生成画像を画面に出力するノードを別々に作成しました.

目次

  1. 概要
  2. パッケージ作成
  3. Publisher:ランダム画像の生成
  4. Subscriber:画像の出力
  5. 動作確認
  6. 参考文献

概要

 ROS2における機能単位はノードです.1ROS2では複数のノードを1つのプロセスで動かすことができ,その実行単位をターゲットと呼びます.ノードひとつひとつに機能を実装することで大きなプロセスを作成することができます.

image.png
※引用 Eureka Box(ユーリカボックス)【体験版】ROS =連載第2回2

 今回のチュートリアルでは画像を生成し,その画像を画面に出力するという機能を実装します(=ターゲット).
 具体的には,ノード間でのトピック通信というメッセージ方式を用いて実装します.トピック通信は非同期・単方向の通信です.メッセージを発信する側と受信する側をそれぞれpublisher, subscriberと呼びます. 

image.png
※引用 Eureka Box(ユーリカボックス)【体験版】ROS =連載第2回2

 今回のチュートリアルでは画像を生成し,その画像を画面に出力するという機能(=ターゲット)を実装したいので,画像生成をpublisherノード,その画像を受け取って画面に出力するノードをsubscriberとして実装しました.

パッケージ作成

 今回のチュートリアル用にパッケージを作成します.
以下のコマンドを実行します.コマンドではパッケージで用いるライブラリなど依存関係を指定することが可能で,

  • rclcpp:ROS2の機能をC++で使用する
  • std_msgs,sensor_msgs:標準的およびセンサー関連のメッセージを扱う
  • cv_bridge,image_transportは画像データを効率的に変換および転送
  • OpenCV:画像の生成および操作

これらのライブラリを指定しています.

cd ~/ros2_ws/src
~/ros2_ws/src$ ros2 pkg create my_opencv_demo --dependencies rclcpp std_msgs sensor_msgs cv_bridge image_transport OpenCV

このコマンド実行後,以下のように新たなディレクトリとファイルが作成されます.

ros2_ws/
├── build/            # ビルドされたファイルが格納される
├── install/          # インストールされたファイルが格納される
├── log/              # ビルドのログが格納される
└── src/              # ソースファイルが格納される
    └── my_opencv_demo/    # 新たに作成されたディレクトリ・ファイル
        ├── CMakeLists.txt      # CMakeのビルド設定ファイル
        ├── package.xml         # パッケージのメタデータと依存関係
        ├── include/            # C++ヘッダーファイル用
        │   └── my_opencv_demo/
        ├── src/                # パッケージのソースファイル用 →ここにノードのソースコードを置く
        └── launch/             # launchファイル用

これでパッケージ作成の準備が整いました.

Publiseher:ランダム画像の生成

 画像を生成し,publishするノードを作成していきます.概要にも述べたとおり,このノードは各ピクセルがランダムカラーの画像を500msで生成・パブリッシュします.

ソースコードは以下の折りたたみ部分をご確認ください.MinimalImagePublisherクラスがrclcpp::Nodeを継承しています.このノードクラスでは定期的にランダムカラーの画像を生成し,パブリッシュするノードを定義しています.画像はOpenCVで生成され、cv_bridgeを通じてROS2のイメージメッセージに変換されてパブリッシュされます.

ソースコード
image_generate_pub.cpp
#include "rclcpp/rclcpp.hpp"
#include "sensor_msgs/msg/image.hpp"
#include "std_msgs/msg/header.hpp"
#include <chrono>
#include <cv_bridge/cv_bridge.h> // cv_bridge converts between ROS 2 image messages and OpenCV image representations.
#include <image_transport/image_transport.hpp> // Using image_transport allows us to publish and subscribe to compressed image streams in ROS2
#include <opencv2/opencv.hpp> // We include everything about OpenCV as we don't care much about compilation time at the moment.

using namespace std::chrono_literals;

class MinimalImagePublisher : public rclcpp::Node {
public:
  MinimalImagePublisher() : Node("opencv_image_publisher"), count_(0) {
    publisher_ =
        this->create_publisher<sensor_msgs::msg::Image>("random_image", 10);
    timer_ = this->create_wall_timer(
        500ms, std::bind(&MinimalImagePublisher::timer_callback, this));
  }

private:
  void timer_callback() {
    // Create a new 640x480 image
    cv::Mat my_image(cv::Size(640, 480), CV_8UC3);

    // Generate an image where each pixel is a random color
    cv::randu(my_image, cv::Scalar(0, 0, 0), cv::Scalar(255, 255, 255));

    // Write message to be sent. Member function toImageMsg() converts a CvImage
    // into a ROS image message
    // Declare the text position
    cv::Point text_position(15, 40);
    cv::Point text_position_2(15, 90);
    // Declare the size and color of the font
    int font_size = 1;
    cv::Scalar font_color(255, 255, 255);

    // Declare the font weight
    int font_weight = 2;

    // Put the text in the image
    cv::putText(my_image, "ROS2 + OpenCV Test Image", text_position, cv::FONT_HERSHEY_COMPLEX, font_size, font_color, font_weight);
    cv::putText(my_image, "Test Image : GxP GxP GxP", text_position_2, cv::FONT_HERSHEY_COMPLEX, font_size, font_color, font_weight);

    msg_ = cv_bridge::CvImage(std_msgs::msg::Header(), "bgr8", my_image)
               .toImageMsg();

    // Publish the image to the topic defined in the publisher
    publisher_->publish(*msg_.get());
    RCLCPP_INFO(this->get_logger(), "Image %ld published", count_);
    count_++;
  }
  rclcpp::TimerBase::SharedPtr timer_;
  sensor_msgs::msg::Image::SharedPtr msg_;
  rclcpp::Publisher<sensor_msgs::msg::Image>::SharedPtr publisher_;
  size_t count_;
};

int main(int argc, char *argv[]) {
  rclcpp::init(argc, argv);
  // create a ros2 node
  auto node = std::make_shared<MinimalImagePublisher>();

  // process ros2 callbacks until receiving a SIGINT (ctrl-c)
  rclcpp::spin(node);
  rclcpp::shutdown();
  return 0;
}

Subscribe:画像の出力

トピック通信によって,Publisherが発信した画像のメッセージを受け取って画面に表示するノードを実装します.

ソースコード
image_sub_view.cpp
#include "rclcpp/rclcpp.hpp"
#include "sensor_msgs/msg/image.hpp"
#include <cv_bridge/cv_bridge.h>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgproc/imgproc.hpp>

class MinimalImageSubscriber : public rclcpp::Node
{
public:
  MinimalImageSubscriber() : Node("opencv_image_subscriber")
  {
    // Initialize subscriber
    subscriber_ = this->create_subscription<sensor_msgs::msg::Image>(
      "random_image", 10, std::bind(&MinimalImageSubscriber::image_callback, this, std::placeholders::_1));
  }

private:
  void image_callback(const sensor_msgs::msg::Image::SharedPtr msg)
  {
    try {
      // Convert ROS image message to OpenCV image
      cv::Mat cv_image = cv_bridge::toCvCopy(msg, "bgr8")->image;

      // Display image
      cv::imshow("Subscribed Image", cv_image);
      cv::waitKey(1); // Wait for a key press for 1 millisecond to allow UI events to process
    } catch (cv_bridge::Exception& e) {
      RCLCPP_ERROR(this->get_logger(), "Could not convert from '%s' to 'bgr8'.", msg->encoding.c_str());
    }
  }

  rclcpp::Subscription<sensor_msgs::msg::Image>::SharedPtr subscriber_;
};

int main(int argc, char *argv[])
{
  rclcpp::init(argc, argv);
  auto node = std::make_shared<MinimalImageSubscriber>();
  rclcpp::spin(node);
  rclcpp::shutdown();
  return 0;
}

cv::Mat cv_image = cv_bridge::toCvCopy(msg, "bgr8")->image;

この部分ではcv_bridge::toCvCopy(msg, "bgr8") 関数が,受け取ったメッセージをOpenCVのcv::Matオブジェクトに変換しています.この変換によってOpenCVを用いて他画像処理を受け取ったノードが自由に行えるようになります.

cv::imshow("Subscribed Image", cv_image);

上記の部分で画像を画面に表示しています.今後,内部で画像変換を行ったり,加工を加えた画像をさらに外部のノードに渡したりすることも考えています.詳細は折りたたみのソースコードをご確認ください.

動作確認

まずは作成したパッケージをコンパイル・ビルドします.

~/ros2_ws$ colcon build --packages-select my_opencv_demo
Starting >>> my_opencv_demo
Finished <<< my_opencv_demo [6.93s]                     

Summary: 1 package finished [7.31s]

無事にビルドが完了しました.次にROS2ワークスペースの環境をセットアップします.このコマンドによって,ビルドされたパッケージが利用可能になります.

~/ros2_ws$ source ~/ros2_ws/install/setup.bash

実際に二つのノードを起動してプロセスを開始してみましょう.以下のコマンドを実行します.こちらは画像を生成するpublisherノードでした.このノードを起動することで生成した画像がpublishされます.

~/ros2_ws$ ros2 run my_opencv_demo minimal_opencv_ros2_node
 // minimal_opencv_ros2_nodeは,画像を生成するノード.

次に,ターミナルで他のセッションを開始して,subscriberノードも起動します.先ほどのpubliserノードとトピック通信がうまくできていれば画像が表示されるはずです.

~/ros2_ws$ ros2 run my_opencv_demo minimal_opencv_ros2_subscriber
 // minimal_opencv_ros2_nodeは,画像を生成するノード.

 無事に表示されました.画像は毎秒生成されてpublishされるのでストリーミングデータですが,新たな画像を受け取って表示画像を更新できていることがわかります.

スクリーンショット 2024-04-19 14.41.27.png

参考文献

  1. 第2回 ROSとROS2のプログラミング概念|コラム

  2. https://member.eureka-box.com/products/4/categories/2148121491/posts/2152934304 2

3
3
1

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
3
3

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?