首页
学习
活动
专区
工具
TVP
发布
社区首页 >问答首页 >安卓:使用OpenCV对齐图像

安卓:使用OpenCV对齐图像
EN

Stack Overflow用户
提问于 2017-06-06 15:26:33
回答 1查看 1.3K关注 0票数 1

我正在尝试使用OpenCV对齐两个图像。我根据我找到的C++/Python教程编写了这段代码:http://www.learnopencv.com/image-alignment-ecc-in-opencv-c-python/

android中的findTransformECC()函数需要一个额外的inputMask参数。C++和Python函数没有这个功能。

我的代码:

import android.graphics.Bitmap;

import org.opencv.android.Utils;
import org.opencv.core.CvType;
import org.opencv.core.Mat;
import org.opencv.core.TermCriteria;
import org.opencv.imgproc.Imgproc;

import static org.opencv.core.CvType.CV_32F;
import static org.opencv.video.Video.MOTION_EUCLIDEAN;
import static org.opencv.video.Video.findTransformECC;

public class ImageProcessor {

    public static Bitmap alignImages(Bitmap A, Bitmap B){
        final int warp_mode = MOTION_EUCLIDEAN;
        Mat matA = new Mat(A.getHeight(),A.getWidth(), CvType.CV_8UC3);
        Mat matAgray = new Mat(A.getHeight(),A.getWidth(), CvType.CV_8U);
        Mat matB = new Mat(B.getHeight(),B.getWidth(), CvType.CV_8UC3);
        Mat matBgray = new Mat(B.getHeight(),B.getWidth(), CvType.CV_8U);
        Mat matBaligned = new Mat(A.getHeight(),A.getWidth(), CvType.CV_8UC3);
        Mat warpMatrix = Mat.eye(3,3,CV_32F);

        Utils.bitmapToMat(A,matA);
        Utils.bitmapToMat(B,matB);


        Imgproc.cvtColor(matA,matAgray, Imgproc.COLOR_BGR2GRAY);
        Imgproc.cvtColor(matB,matBgray,Imgproc.COLOR_BGR2GRAY);

        int numIter = 5000;
        double terminationEps = 1e-10;
        TermCriteria criteria = new TermCriteria(TermCriteria.COUNT+TermCriteria.EPS,numIter,terminationEps);

        findTransformECC(matAgray,matBgray,warpMatrix,warp_mode,criteria,matBgray);
        Imgproc.warpPerspective(matA,matBaligned,warpMatrix,matA.size(),Imgproc.INTER_LINEAR+ Imgproc.WARP_INVERSE_MAP);
        Bitmap alignedBMP = Bitmap.createBitmap(A.getWidth(),A.getHeight(),null);
        Utils.matToBitmap(matBaligned,alignedBMP);
        return alignedBMP;
    }

}

我得到以下错误

W/System.err: CvException [org.opencv.core.CvException: cv::Exception: /build/master_pack-android/opencv/modules/imgproc/src/imgwarp.cpp:5987: error: (-215) (M0.type() == CV_32F || M0.type() == CV_64F) && M0.rows == 2 && M0.cols == 3 in function void cv::warpAffine(cv::InputArray, cv::OutputArray, cv::InputArray, cv::Size, int, int, const Scalar&)
W/System.err: ]
W/System.err:     at org.opencv.video.Video.findTransformECC_0(Native Method)
W/System.err:     at org.opencv.video.Video.findTransformECC(Video.java:132)
W/System.err:     at com.test.imgptest.ImageProcessor.alignImages(ImageProcessor.java:42)
W/System.err:     at com.test.imgptest.MainActivity.onActivityResult(MainActivity.java:141)
W/System.err:     at android.app.Activity.dispatchActivityResult(Activity.java:6937)
W/System.err:     at android.app.ActivityThread.deliverResults(ActivityThread.java:4122)
W/System.err:     at android.app.ActivityThread.handleSendResult(ActivityThread.java:4169)
W/System.err:     at android.app.ActivityThread.-wrap20(ActivityThread.java)
W/System.err:     at android.app.ActivityThread$H.handleMessage(ActivityThread.java:1552)
W/System.err:     at android.os.Handler.dispatchMessage(Handler.java:102)
W/System.err:     at android.os.Looper.loop(Looper.java:154)
W/System.err:     at android.app.ActivityThread.main(ActivityThread.java:6186)
W/System.err:     at java.lang.reflect.Method.invoke(Native Method)
W/System.err:     at com.android.internal.os.ZygoteInit$MethodAndArgsCaller.run(ZygoteInit.java:889)
W/System.err:     at com.android.internal.os.ZygoteInit.main(ZygoteInit.java:779)
EN

回答 1

Stack Overflow用户

发布于 2018-07-01 01:23:17

可接受答案中的完整函数:

public static Bitmap alignImagesHomography(Bitmap A, Bitmap B)
{
    final int warp_mode = MOTION_HOMOGRAPHY;
    Mat matA = new Mat(A.getHeight(), A.getWidth(), CvType.CV_8UC3);
    Mat matAgray = new Mat(A.getHeight(), A.getWidth(), CvType.CV_8U);
    Mat matB = new Mat(B.getHeight(), B.getWidth(), CvType.CV_8UC3);
    Mat matBgray = new Mat(B.getHeight(), B.getWidth(), CvType.CV_8U);
    Mat matBaligned = new Mat(A.getHeight(), A.getWidth(), CvType.CV_8UC3);
    Mat warpMatrix = Mat.eye(3, 3, CV_32F);
    Utils.bitmapToMat(A, matA);
    Utils.bitmapToMat(B, matB);
    Imgproc.cvtColor(matA, matAgray, Imgproc.COLOR_BGR2GRAY);
    Imgproc.cvtColor(matB, matBgray, Imgproc.COLOR_BGR2GRAY);
    int numIter = 5;
    double terminationEps = 1e-10;
    TermCriteria criteria = new TermCriteria(TermCriteria.COUNT + TermCriteria.EPS, numIter, terminationEps);
    findTransformECC(matAgray, matBgray, warpMatrix, warp_mode, criteria, matBgray);
    Imgproc.warpPerspective(matB, matBaligned, warpMatrix, matA.size(), Imgproc.INTER_LINEAR + Imgproc.WARP_INVERSE_MAP);
    Bitmap alignedBMP = Bitmap.createBitmap(A.getWidth(), A.getHeight(), Bitmap.Config.RGB_565);
    Utils.matToBitmap(matBaligned, alignedBMP);
    return alignedBMP;
}

public static Bitmap alignImagesEuclidean(Bitmap A, Bitmap B)
{
    final int warp_mode = MOTION_EUCLIDEAN;
    Mat matA = new Mat(A.getHeight(), A.getWidth(), CvType.CV_8UC3);
    Mat matAgray = new Mat(A.getHeight(), A.getWidth(), CvType.CV_8U);
    Mat matB = new Mat(B.getHeight(), B.getWidth(), CvType.CV_8UC3);
    Mat matBgray = new Mat(B.getHeight(), B.getWidth(), CvType.CV_8U);
    Mat matBaligned = new Mat(A.getHeight(), A.getWidth(), CvType.CV_8UC3);
    Mat warpMatrix = Mat.eye(2,3,CV_32F);
    Utils.bitmapToMat(A, matA);
    Utils.bitmapToMat(B, matB);
    Imgproc.cvtColor(matA, matAgray, Imgproc.COLOR_BGR2GRAY);
    Imgproc.cvtColor(matB, matBgray, Imgproc.COLOR_BGR2GRAY);
    int numIter = 5;
    double terminationEps = 1e-10;
    TermCriteria criteria = new TermCriteria(TermCriteria.COUNT + TermCriteria.EPS, numIter, terminationEps);
    findTransformECC(matAgray, matBgray, warpMatrix, warp_mode, criteria, matBgray);
    Imgproc.warpAffine(matB, matBaligned, warpMatrix, matA.size(), Imgproc.INTER_LINEAR + Imgproc.WARP_INVERSE_MAP);
    Bitmap alignedBMP = Bitmap.createBitmap(A.getWidth(), A.getHeight(), Bitmap.Config.RGB_565);
    Utils.matToBitmap(matBaligned, alignedBMP);
    return alignedBMP;
}

下面是一个使用alignMTB的更快的版本:

public static Bitmap AlignExposures(Mat[] image_list) {
    List<Mat> src = Arrays.asList(image_list);
    Bitmap output = Bitmap.createBitmap(bitmapWidth,bitmapHeight, Bitmap.Config.ARGB_8888);
    AlignMTB align = createAlignMTB(8, 4, false);
    align.process(src,src);
    for(int i = 1; i < src.size(); i++) {
        addWeighted(src.get(0),1-1/(i+1),src.get(i),1/(i+1),0,src.get(0));
    }
    Utils.matToBitmap(src.get(0),output);
    return output;
}
票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/44383801

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档