Android で OpenCV と　MobleNet を使って物体検出する #Android

AndroidStudio で OpenCV をインポートする
の続きです。

MobleNet

MoblieNetは、モバイル用に軽量化された　Neural Network　のフレームワークです。

OpenCV の MobleNet

OpenCV 3.3 から対応している。
cv::dnn モジュールと Java ラッパーが用意されている。

opencv : Deep Neural Networks (dnn module)

opencv : Class Dnn

mobilenet-objdetect

OpenCVのレポジトリにある Android 用のサンプルコードです。
今回はこれを試す。

学習済みデータ

上記のサンプルコードには、学習済みデータとして、
下記の２つのファイルが必要です。
- 構成ファイル MobileNetSSD_deploy.prototxt
- 重みファイル MobileNetSSD_deploy.caffemodel

下記からダウンロードする。
https://github.com/chuanqi305/MobileNet-SSD

検出できる物体は下記の20種類。

aeroplane (飛行機)
bicycle (自転車)
bird (鳥)
boat (小型船)
bottle (瓶)
bus (バス)
car (車)
cat (猫)
chair (椅子)
cow (乳牛)
diningtable (食卓)
dog (犬)
horse (馬)
motorbike (オートバイ)
person (人物)
pottedplant (鉢植え)
sheep (羊)
sofa (長椅子)
train (列車)
tvmonitor (テレビ受像機)

アプリを作成する

まず、下記を読んでください。
Android　で OpenCV のサンプル camerapreview を試す

CameraActivity を継承して MainActivity.java を作成する。

Neural Network を取得する。

public class MainActivity extends CameraActivity implements CvCameraViewListener2 {

public void onCameraViewStarted(int width, int height) {
    // 学習済みモデルを設定する
    String proto = getPath("MobileNetSSD_deploy.prototxt", this);
    String weights = getPath("MobileNetSSD_deploy.caffemodel", this);
    // Neural Network を取得する
    mNet = Dnn.readNetFromCaffe(proto, weights);
}

private  String getPath(String fileName, Context context) {
        // Assets フォルダーのファイルを外部記憶にコピーする
        // 詳細 略

物体を検出する。

public Mat onCameraFrame(CvCameraViewFrame inputFrame) {
    final int IN_WIDTH = 300;
    final int IN_HEIGHT = 300;
    final double MEAN_VAL = 127.5;

    Mat frame = inputFrame.rgba();

    // 物体検出する物体の大きさを設定する
    Mat blob = Dnn.blobFromImage(frame,         IN_SCALE_FACTOR,
     new Size(IN_WIDTH, IN_HEIGHT),
    new Scalar(MEAN_VAL, MEAN_VAL, MEAN_VAL), 
    /*swapRB*/false, /*crop*/false);
    mNet.setInput(blob);

    // フォワードパスを実行し、物体を検出する
    Mat detections = mNet.forward();