C#でOpenCvSharp3のGrabCutを試す #C#

はじめに

C#でOpenCvSharp3というライブラリを使ってGrabCutを試しました。
ネットにC#でOpenCVを扱った記事も少ないため、備忘録として残します。
OpenCVについては素人以下なので色々間違っているかも知れません。

環境

Windows 8.1
Visual Studio 2015

プロジェクト作成

Visual Studioを起動
C#コンソールアプリケーションを新規作成
プロジェクト名は取りあえずTestGrabCutとしました
NuGetでOpenCvSharp3-AnyCPUを追加

コード

Program.csを開き、冒頭にusing OpenCvSharpを追加します。
プロジェクトルートにGrabCutをしたい画像を追加し、プロパティの出力ディレクトリにコピーを常にコピーするに設定。
今回はこちらの画像を利用させていただきました。

後は以下の通りです。

Program.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using OpenCvSharp;

namespace TestGrabCut
{
    class Program
    {
        static void Main(string[] args)
        {
            // 前景らしい領域を表す定数
            const int GC_PR_FGD = 3;

            // 対象画像オブジェクト
            Mat image = new Mat("iphone8IMGL7576_TP_V4.jpg", ImreadModes.Color);

            // 前景を含む領域
            Rect rectangle = new Rect(300, 20, 650, 500);

            // 前景を抽出
            var result = new Mat(image.Size(), MatType.CV_8U);
            var bgdModel = new Mat();
            var fgdModel = new Mat();
            var source = new Mat(image.Size(), MatType.CV_8U, new Scalar(GC_PR_FGD));
            // GrabCutするイテレーションに5回を指定
            Cv2.GrabCut(image, result, rectangle, bgdModel, fgdModel, 5, GrabCutModes.InitWithRect);
            Cv2.Compare(result, source, result, CmpTypes.EQ);
            // 抽出した前景を貼り付ける画像、ソースと同サイズの真っ白な画像を指定
            Mat foreground = new Mat(image.Size(), MatType.CV_8UC3, new Scalar(255, 255, 255));
            image.CopyTo(foreground, result);

            // 表示
            Cv2.ImShow("タイトル", foreground);
            Cv2.WaitKey();
        }
    }
}