Java版OpenCVでラベリングした画像の色を変えたい

前提・実現したいこと

Java版OpenCV3.4.2で二値画像にラベリングをし、連結成分の面積が一番大きな部分以外を黒で塗りつぶしたいです(切り抜き)。
connectedComponentsWithStatsを用いて面積情報を取得するところまではできたのですが
指定したラベルの連結成分の色を変えることができずに困っています。

該当のソースコード

Java
1 public void getframe(Mat src_img, Mat dst_img){
2
3        int rows = src_img.width();
4        int cols = src_img.height();
5
6//二値化
7        Mat gray_img = new Mat(cols,rows, CvType.CV_8UC1);
8        Imgproc.cvtColor(src_img,gray_img,Imgproc.COLOR_RGB2GRAY);
9        int blockSize=Math.min(cols,rows)/40*2+1;
10        Imgproc.adaptiveThreshold(gray_img,dst_img,255,Imgproc.ADAPTIVE_THRESH_GAUSSIAN_C,Imgproc.THRESH_BINARY_INV,blockSize,3);
11
12      
13//ラベリング実行
14 Mat labelImage = new Mat(cols,rows,CvType.CV_32S);
15        Mat stats = new Mat();
16        Mat centroids = new Mat();
17
18        int nLabels = Imgproc.connectedComponentsWithStats(dst_img,labelImage,stats,centroids,8,CvType.CV_32S);
19
20       
21
22//面積情報を取得
23int[] result = new int[nLabels];
24        int max = 0;
25        int count = 0;
26
27        for(int i=1;i<nLabels;i++){
28            stats.get(i,Imgproc.CC_STAT_AREA,result);
29            if(max < result[i]){
30                max = result[i];
31                count = i;
32            }
33        }
34
35//連結成分の中で一番面積が大きいもの以外を黒塗り
36   　　for(int i=1;i<nLabels;i++){
37            if(i != count){
38                //黒で塗りつぶす
39            }
40        }
41
42}

試したこと

c++でのconnectedComponentsにおける指定したラベルの塗りつぶしを参考にしています。

https://github.com/atinfinity/lab/wiki/OpenCV%E3%82%92%E4%BD%BF%E3%81%A3%E3%81%9F%E3%83%A9%E3%83%99%E3%83%AA%E3%83%B3%E3%82%B0

しかし、Java版OpenCVに画素値を変更するVec3bが無いようで、
代わりに指定した座標の画素値を取り出す「mat.get()」と
指定した座標の画素値を変更する「mat.put()」を利用しようと試みたのですがどうにもうまくいきません。

そもそも参考にしているサイトにおいて

std::vector<cv::Vec3b> colors(nLabels);

というVec3b型の可変長配列を作っているところまでは理解できるのですが

colors[0] = cv::Vec3b(0, 0, 0);

上記のようにcolols[ラベルの番号]だけで指定したラベルにアクセスできているというのが理解できません。
colorsはあくまでnLabels個のVec3b型の配列として定義されただけなのに
なぜラベル番号を引数に入れるだけで該当ラベルの画素値を操作できるようになるのでしょうか。

伝わりづらいしし質問で申し訳ありませんがご教授いただけると幸いです。

c++
1std::vector<cv::Vec3b> colors(nLabels);
2    colors[0] = cv::Vec3b(0, 0, 0);
3    for(int label = 1; label < nLabels; ++label)
4    {
5        colors[label] = cv::Vec3b((rand()&255), (rand()&255), (rand()&255));
6    }
7
8    // ラベリング結果の描画
9    cv::Mat dst(src.size(), CV_8UC3);
10    for(int y = 0; y < dst.rows; ++y)
11    {
12        for(int x = 0; x < dst.cols; ++x)
13        {
14            int label = labelImage.at<int>(y, x);
15            cv::Vec3b &pixel = dst.at<cv::Vec3b>(y, x);
16            pixel = colors[label];
17        }
18
19

回答2件

ベストアンサー

上記のようにcolols[ラベルの番号]だけで指定したラベルにアクセスできているというのが理解できません。

C++
1std::vector<cv::Vec3b> colors;

と宣言した場合は空の配列なので、いきなりcolors[0]と指定はできませんが、

C++
1std::vector<cv::Vec3b> colors(n);

とn個の要素を用意する指定があるためにcolors[0]～colors[n-1]にアクセスできます。

参考：C++ 動的配列クラス std::vector 入門
4-2.要素数を指定した宣言を参照。

指定したラベルにアクセスできるようにしているのは、以下のコードです。

C++
1cv::Vec3b &pixel = dst.at<cv::Vec3b>(y, x);

ここの&は参照型といい、この型で宣言した変数に値の代入すると、参照元もその値に変更されるという便利な型なのです。
参考：参照 (C++)
C++入門：3章 C++の新しい文法-> 参照型

投稿2018/07/30 21:01

編集2018/07/30 21:30

phoenix3430025

総合スコア135

参照型について学ぶことができましたありがとうございました。

指定したラベルの色を変えるということはコードを書き換えた結果成功しました。
連結成分の面積(CC_STAS_AREA)を取得する際に

     stats.get(i,Imgproc.CC_STAT_AREA,result);
            if(max < result[i]){
                max = result[i];
                count = i;
            }

としていましたが

     stats.get(i,Imgproc.CC_STAT_AREA,result);
            if(max < result[0]){
                max = result[0];
                count = i;
            }

とすることで正しく連結成分の面積情報を取得することができました

投稿2018/07/31 19:02

jannema

総合スコア9