1. 程式人生 > >Android 使用OPENCV實現影象實時對比

Android 使用OPENCV實現影象實時對比

本文主要介紹使用OpenCV實現相機實時對比圖片,得到相似度,可用於實現類似ar紅包的功能
首先下載SDK,OpenCV-3.2.0(下載官網:http://opencv.org),點選OpenCV for Android下載,下載完成後解壓
SDK中包含很多有趣的demo,可以一併看一下,所以,最好新建一個Eclipse的工作空間

1. 建立demo工程

新建一個Android Application,命名為OpenCVDemo

2. 整合OpenCV SDK

將完整的OpenCV SDK拷貝到工作空間中,將OpenCV_SDK_3.2.0/sdk/java匯入到Eclipse中,這個就是我們需要整合的library,將它與我們的demo關聯

3. 配置SDK

●因為要用到相機,所以在AndroidManifest.xml新增以下程式碼

新增許可權

<uses-permission android:name="android.permission.CAMERA"/>  
<uses-feature android:name="android.hardware.camera" android:required="false"/>    
<uses-feature android:name="android.hardware.camera.autofocus" android:required
="false"/>
<uses-feature android:name="android.hardware.camera.front" android:required="false"/> <uses-feature android:name="android.hardware.camera.front.autofocus" android:required="false"/>

在application節點下新增

android:theme="@android:style/Theme.NoTitleBar.Fullscreen"
●在MainActivity中繼承CvCameraViewListener2介面

CvCameraViewListener2是使用OpenCV相機的核心監聽

public class MainActivity extends Activity implements CvCameraViewListener2 {
    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_main);
    }
    @Override
    public void onCameraViewStarted(int width, int height) {
        // TODO Auto-generated method stub

    }
    @Override
    public void onCameraViewStopped() {
        // TODO Auto-generated method stub

    }
    @Override
    public Mat onCameraFrame(CvCameraViewFrame inputFrame) {
        // TODO Auto-generated method stub
        return null;
    }
}

onCameraViewStarted:相機啟動時呼叫
onCameraViewStopped:相機銷燬時呼叫
onCameraFrame: 相機工作時呼叫,引數是相機每一幀的影象,實時對比就在這個方法中進行

●初始化相機

編寫佈局檔案,新增相機控制元件,相機控制元件實質是SurfaceView

<RelativeLayout xmlns:android="http://schemas.android.com/apk/res/android"
    xmlns:tools="http://schemas.android.com/tools"
    android:layout_width="match_parent"
    android:layout_height="match_parent"
    tools:context="com.example.opencvtest.MainActivity" >

    <org.opencv.android.JavaCameraView
        android:id="@+id/cv"
        android:layout_width="fill_parent"
        android:layout_height="fill_parent" />
</RelativeLayout>

在OnCreate中初始化相機,並新增開啟相機的回撥

    /**
     * CV相機
     */
    private CameraBridgeViewBase mCVCamera;
    /**
     * 載入OpenCV的回撥
     */
    private BaseLoaderCallback mLoaderCallback;

    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_main);
        // 初始化CV相機
        mCVCamera = (CameraBridgeViewBase) findViewById(R.id.cv);
        mCVCamera.setVisibility(CameraBridgeViewBase.VISIBLE);
        // 設定相機監聽
        mCVCamera.setCvCameraViewListener(this);
        // 連線到OpenCV的回撥
        mLoaderCallback = new BaseLoaderCallback(this) {
            @Override
            public void onManagerConnected(int status) {
                switch (status) {
                case LoaderCallbackInterface.SUCCESS:
                    mCVCamera.enableView();
                    break;
                default:
                    break;
                }
            }
        };
    }

介面載入完成時說明OpenCV已經初始化完成,所以重寫onResume()方法,開啟相機

    @Override
    protected void onResume() {
        // 介面載入完成的時候向OpenCV的連接回調發送連線成功的訊號
        if (OpenCVLoader.initDebug()) {
            mLoaderCallback.onManagerConnected(LoaderCallbackInterface.SUCCESS);
        }
        super.onResume();
    }

重寫onPause方法,及時銷燬相機

    @Override
    protected void onPause() {
        super.onPause();
        // 銷燬OpenCV相機
        if (mCVCamera != null)
            mCVCamera.disableView();
    }

編輯onCameraFrame方法,讓相機的每一幀展示在介面上

    @Override
    public Mat onCameraFrame(CvCameraViewFrame inputFrame) {
        return inputFrame.rgba();
    }

到這開啟相機的程式碼已經完成了,但是此時的工程需要依賴SDK中提供的OpenCV_3.2.0_Manager才能執行,這顯然是不滿足多數專案需求的,需要進行以下操作:
為專案新增C/C++編譯支援,右鍵專案,選擇Android Tools–>Add Native Support,點選finish
開啟jni/Android.mk

在紅框處新增以下程式碼

OPENCV_CAMERA_MODULES:=on
OPENCV_INSTALL_MODULES:=on
OPENCV_LIB_TYPE:=SHARED
ifdef OPENCV_ANDROID_SDK
  ifneq ("","$(wildcard $(OPENCV_ANDROID_SDK)/OpenCV.mk)")
    include ${OPENCV_ANDROID_SDK}/OpenCV.mk
  else
    include ${OPENCV_ANDROID_SDK}/sdk/native/jni/OpenCV.mk
  endif
else
  include ../OpenCV-android-sdk/sdk/native/jni/OpenCV.mk
endif

因為include使用的是相對路徑,所以最好將OpenCVDemo和OpenCV-android-sdk放在同一個目錄下,否則需要根據自己目錄結構修改相對路徑

● 執行

4. 編寫圖片對比的方法

新建一個HistUtils.java,新增下面兩個方法

    /**
     * 比較來個矩陣的相似度
     * 
     * @param mBitmap1
     * @param mBitmap2
     * @return
     */
    public static double comPareHist(Bitmap mBitmap1, Bitmap mBitmap2) {
        Mat mat1 = new Mat();
        Mat mat2 = new Mat();
        Utils.bitmapToMat(mBitmap1, mat1);
        Utils.bitmapToMat(mBitmap2, mat2);
        return comPareHist(mat1, mat2);
    }

    /**
     * 比較來個矩陣的相似度
     * 
     * @param mat1
     * @param mat2
     * @return
     */
    public static double comPareHist(Mat mat1, Mat mat2) {
        Mat srcMat = new Mat();
        Mat desMat = new Mat();
        Imgproc.cvtColor(mat1, srcMat, Imgproc.COLOR_BGR2GRAY);
        Imgproc.cvtColor(mat2, desMat, Imgproc.COLOR_BGR2GRAY);
        srcMat.convertTo(srcMat, CvType.CV_32F);
        desMat.convertTo(desMat, CvType.CV_32F);
        double target = Imgproc.compareHist(srcMat, desMat,
                Imgproc.CV_COMP_CORREL);
        return target;
    }

Imgproc.compareHist(Mat H1, Mat H2, int method)這個方法是OpenCV提供的通過直方圖演算法對比兩張圖片的相似度,完全相同的兩張圖片相似度為1。

新增對比按鈕,實現主動對比和自動對比

核心的程式碼上面已經寫完了,剩下的就是新增按鈕和觸發對比,就不詳細介紹了,直接把完整的程式碼附上,備註很詳細

MainActivity.java

package com.example.opencvdemo;

import org.opencv.android.BaseLoaderCallback;
import org.opencv.android.CameraBridgeViewBase;
import org.opencv.android.CameraBridgeViewBase.CvCameraViewFrame;
import org.opencv.android.CameraBridgeViewBase.CvCameraViewListener2;
import org.opencv.android.LoaderCallbackInterface;
import org.opencv.android.OpenCVLoader;
import org.opencv.android.Utils;
import org.opencv.core.Mat;

import android.annotation.SuppressLint;
import android.app.Activity;
import android.graphics.Bitmap;
import android.graphics.Bitmap.Config;
import android.os.Bundle;
import android.os.Handler;
import android.os.Message;
import android.view.View;
import android.view.View.OnClickListener;
import android.widget.Button;
import android.widget.ImageView;
import android.widget.TextView;

public class MainActivity extends Activity implements CvCameraViewListener2 {

    /**
     * 展示原圖和對比圖
     */
    private ImageView iv1, iv2;

    /**
     * 原圖和對比圖
     */
    private Bitmap bmp1, bmp2;

    /**
     * 拍攝原圖、拍攝對比圖、對比、自動對比
     */
    private Button pz1, pz2, db, zddb;

    /**
     * 顯示相似度(完全相同值為1)
     */
    private TextView tv;

    /**
     * CV相機
     */
    private CameraBridgeViewBase mCVCamera;

    /**
     * 載入OpenCV的回撥
     */
    private BaseLoaderCallback mLoaderCallback;

    /**
     * 拍照狀態 0:不拍照 ,1:拍原圖,2:拍對比圖,3:拍對比圖並自動對比
     */
    private int isTakePhoto = 0;

    /**
     * 用於定時執行圖片對比
     */
    private Handler handler;

    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_main);
        // 初始化
        init();
    }

    @SuppressLint("HandlerLeak")
    private void init() {
        // 初始化
        iv1 = (ImageView) findViewById(R.id.iv1);
        iv2 = (ImageView) findViewById(R.id.iv2);
        pz1 = (Button) findViewById(R.id.pz1);
        pz2 = (Button) findViewById(R.id.pz2);
        db = (Button) findViewById(R.id.db);
        zddb = (Button) findViewById(R.id.zddb);
        tv = (TextView) findViewById(R.id.tv);

        handler = new Handler() {
            @Override
            public void handleMessage(Message msg) {
                super.handleMessage(msg);
                // 自動拍照並對比
                isTakePhoto = 3;
            }
        };

        OnClickListener btnOnClick = new OnClickListener() {
            @Override
            public void onClick(View v) {
                switch (v.getId()) {
                case R.id.pz1:
                    // 拍原圖
                    isTakePhoto = 1;
                    break;
                case R.id.pz2:
                    // 拍對比圖
                    isTakePhoto = 2;
                    break;
                case R.id.db:
                    hist();
                    break;
                case R.id.zddb:
                    // 拍對比圖並自動對比
                    isTakePhoto = 3;
                    break;
                }
            }
        };
        // 設定點選事件
        pz1.setOnClickListener(btnOnClick);
        pz2.setOnClickListener(btnOnClick);
        db.setOnClickListener(btnOnClick);
        zddb.setOnClickListener(btnOnClick);

        // 初始化CV相機
        mCVCamera = (CameraBridgeViewBase) findViewById(R.id.cv);
        mCVCamera.setVisibility(CameraBridgeViewBase.VISIBLE);
        // 設定相機監聽
        mCVCamera.setCvCameraViewListener(this);

        // 連線到OpenCV的回撥
        mLoaderCallback = new BaseLoaderCallback(this) {
            @Override
            public void onManagerConnected(int status) {
                switch (status) {
                case LoaderCallbackInterface.SUCCESS:
                    mCVCamera.enableView();
                    break;
                default:
                    break;
                }
            }
        };
    }

    private void hist() {
        // 對比演算法會耗時,導致頁面卡頓,所以新開執行緒進行對比
        new Thread(new Runnable() {
            @Override
            public void run() {
                // 對比
                final double target = HistUtils.comPareHist(bmp1, bmp2);
                MainActivity.this.runOnUiThread(new Runnable() {
                    @Override
                    public void run() {
                        // 將相似度顯示在左上角
                        tv.setText("相似度:" + target);
                    }
                });
            }
        }).start();
    }

    @Override
    public void onCameraViewStarted(int width, int height) {
    }

    @Override
    public void onCameraViewStopped() {
    }

    @Override
    public Mat onCameraFrame(CvCameraViewFrame inputFrame) {// 相機拍攝每一幀的影象,都在此處理
        // 獲取相機中的影象
        final Mat rgba = inputFrame.rgba();
        if (isTakePhoto != 0) {

            // 記錄拍照狀態
            final int who = isTakePhoto;
            // 重置拍照狀態
            isTakePhoto = 0;

            // 要把Mat物件轉換成Bitmap物件,需要建立一個寬高相同的Bitmap物件昨晚參數
            final Bitmap bmp = Bitmap.createBitmap(rgba.cols(), rgba.rows(),
                    Config.RGB_565);

            // 記錄要展示圖片的ImageView
            ImageView iv = null;
            // Mat >>> Bitmap
            Utils.matToBitmap(rgba, bmp);

            if (who == 1) {
                // 展示原圖
                iv = iv1;
                bmp1 = bmp;
            } else if (who == 2) {
                // 展示對比圖
                iv = iv2;
                bmp2 = bmp;
            } else {
                // 展示對比圖
                iv = iv2;
                bmp2 = bmp;
                // 對比
                hist();
                // 每隔0.5秒對比一次
                handler.sendEmptyMessageDelayed(1, 500);
            }
            // 記錄要展示圖片的ImageView
            final ImageView image = iv;
            runOnUiThread(new Runnable() {
                @Override
                public void run() {
                    // 展示拍到的圖片
                    image.setImageBitmap(bmp);
                    if (bmp1 != null) {
                        // 如果原圖已經拍好了,那麼可以進行自動對比,將自動對比按鈕設定為可用
                        zddb.setEnabled(true);
                        if (bmp2 != null) {
                            // 如果原圖和對比圖都已經拍好了,那麼可以進行對比,將對比按鈕設定為可用
                            db.setEnabled(true);
                        }
                    }
                }
            });
        }
        // 將每一幀的影象展示在介面上
        return inputFrame.rgba();
    }

    @Override
    protected void onResume() {
        // 介面載入完成的時候向OpenCV的連接回調發送連線成功的訊號
        if (OpenCVLoader.initDebug()) {
            mLoaderCallback.onManagerConnected(LoaderCallbackInterface.SUCCESS);
        }
        super.onResume();
    }

    @Override
    protected void onPause() {
        super.onPause();
        // 銷燬OpenCV相機
        if (mCVCamera != null)
            mCVCamera.disableView();
    }

}
<RelativeLayout xmlns:android="http://schemas.android.com/apk/res/android"
    xmlns:tools="http://schemas.android.com/tools"
    android:layout_width="match_parent"
    android:layout_height="match_parent"
    tools:context="com.example.opencvdemo.MainActivity" >

    <RelativeLayout
        android:id="@+id/rl"
        android:layout_width="match_parent"
        android:layout_height="match_parent" >

        <org.opencv.android.JavaCameraView
            android:id="@+id/cv"
            android:layout_width="fill_parent"
            android:layout_height="fill_parent" />

        <TextView
            android:id="@+id/tv"
            android:layout_width="wrap_content"
            android:layout_height="wrap_content"
            android:background="#ffffff"
            android:textColor="#000000"
            android:textSize="13sp" />
    </RelativeLayout>

    <LinearLayout
        android:layout_width="wrap_content"
        android:layout_height="wrap_content"
        android:layout_centerInParent="true"
        android:orientation="horizontal" >

        <ImageView
            android:id="@+id/iv1"
            android:layout_width="150dp"
            android:layout_height="wrap_content"
            android:src="@drawable/red" />

        <ImageView
            android:id="@+id/iv2"
            android:layout_width="150dp"
            android:layout_height="wrap_content"
            android:src="@drawable/red" />
    </LinearLayout>

    <LinearLayout
        android:id="@+id/ll"
        android:layout_width="match_parent"
        android:layout_height="wrap_content"
        android:layout_alignParentBottom="true"
        android:orientation="horizontal" >

        <Button
            android:id="@+id/pz1"
            android:layout_width="0dp"
            android:layout_height="wrap_content"
            android:layout_weight="1"
            android:text="拍照1" />

        <Button
            android:id="@+id/pz2"
            android:layout_width="0dp"
            android:layout_height="wrap_content"
            android:layout_weight="1"
            android:text="拍照2" />

        <Button
            android:id="@+id/db"
            android:layout_width="0dp"
            android:layout_height="wrap_content"
            android:layout_weight="1"
            android:enabled="false"
            android:text="對比" />

        <Button
            android:id="@+id/zddb"
            android:layout_width="0dp"
            android:layout_height="wrap_content"
            android:layout_weight="1"
            android:enabled="false"
            android:text="自動對比" />
    </LinearLayout>

</RelativeLayout>

red.png(自己做的圖片,有點醜)
drawable-hdpi/red.png

● 執行

這裡寫圖片描述

● 完成,歡迎吐槽