Add ocl::stereobp function.

OpenCL StereoBeliefPropagation, ported from GPU implementation.

modules/ocl/include/opencv2/ocl/ocl.hpp

@@ -1701,6 +1701,73 @@ namespace cv
         private:
             oclMat minSSD, leBuf, riBuf;
         };
+        class CV_EXPORTS StereoBeliefPropagation
+        {
+        public:
+            enum { DEFAULT_NDISP  = 64 };
+            enum { DEFAULT_ITERS  = 5  };
+            enum { DEFAULT_LEVELS = 5  };
+            static void estimateRecommendedParams(int width, int height, int &ndisp, int &iters, int &levels);
+            explicit StereoBeliefPropagation(int ndisp  = DEFAULT_NDISP,
+                                             int iters  = DEFAULT_ITERS,
+                                             int levels = DEFAULT_LEVELS,
+                                             int msg_type = CV_16S);
+            StereoBeliefPropagation(int ndisp, int iters, int levels,
+                                    float max_data_term, float data_weight,
+                                    float max_disc_term, float disc_single_jump,
+                                    int msg_type = CV_32F);
+            void operator()(const oclMat &left, const oclMat &right, oclMat &disparity);
+            void operator()(const oclMat &data, oclMat &disparity);
+            int ndisp;
+            int iters;
+            int levels;
+            float max_data_term;
+            float data_weight;
+            float max_disc_term;
+            float disc_single_jump;
+            int msg_type;
+        private:
+            oclMat u, d, l, r, u2, d2, l2, r2;
+            std::vector<oclMat> datas;
+            oclMat out;
+        };
+        class CV_EXPORTS StereoConstantSpaceBP
+        {
+        public:
+            enum { DEFAULT_NDISP    = 128 };
+            enum { DEFAULT_ITERS    = 8   };
+            enum { DEFAULT_LEVELS   = 4   };
+            enum { DEFAULT_NR_PLANE = 4   };
+            static void estimateRecommendedParams(int width, int height, int &ndisp, int &iters, int &levels, int &nr_plane);
+            explicit StereoConstantSpaceBP(int ndisp    = DEFAULT_NDISP,
+                                           int iters    = DEFAULT_ITERS,
+                                           int levels   = DEFAULT_LEVELS,
+                                           int nr_plane = DEFAULT_NR_PLANE,
+                                           int msg_type = CV_32F);
+            StereoConstantSpaceBP(int ndisp, int iters, int levels, int nr_plane,
+                                  float max_data_term, float data_weight, float max_disc_term, float disc_single_jump,
+                                  int min_disp_th = 0,
+                                  int msg_type = CV_32F);
+            void operator()(const oclMat &left, const oclMat &right, oclMat &disparity);
+            int ndisp;
+            int iters;
+            int levels;
+            int nr_plane;
+            float max_data_term;
+            float data_weight;
+            float max_disc_term;
+            float disc_single_jump;
+            int min_disp_th;
+            int msg_type;
+            bool use_local_init_data_cost;
+        private:
+            oclMat u[2], d[2], l[2], r[2];
+            oclMat disp_selected_pyr[2];
+            oclMat data_cost;
+            oclMat data_cost_selected;
+            oclMat temp;
+            oclMat out;
+        };
     }
 }
 #if defined _MSC_VER && _MSC_VER >= 1200

modules/ocl/test/test_calib3d.cpp

@@ -15,7 +15,7 @@
 // Third party copyrights are property of their respective owners.
 //
 // @Authors
-
+//     Peng Xiao, pengxiao@outlook.com
 //
 // Redistribution and use in source and binary forms, with or without modification,
 // are permitted provided that the following conditions are met:
@@ -63,12 +63,12 @@ PARAM_TEST_CASE(StereoMatchBM, int, int)
     }
 };
 
-TEST_P(StereoMatchBM, Accuracy)
+TEST_P(StereoMatchBM, Regression)
 {
 
-    Mat left_image  = readImage(workdir + "../ocl/aloe-L.png", IMREAD_GRAYSCALE);
-    Mat right_image = readImage(workdir + "../ocl/aloe-R.png", IMREAD_GRAYSCALE);
-    Mat disp_gold   = readImage(workdir + "../ocl/aloe-disp.png", IMREAD_GRAYSCALE);
+    Mat left_image  = readImage("stereobm/aloe-L.png", IMREAD_GRAYSCALE);
+    Mat right_image = readImage("stereobm/aloe-R.png", IMREAD_GRAYSCALE);
+    Mat disp_gold   = readImage("stereobm/aloe-disp.png", IMREAD_GRAYSCALE);
 	ocl::oclMat d_left, d_right;
 	ocl::oclMat d_disp(left_image.size(), CV_8U);
 	Mat  disp;
@@ -88,7 +88,50 @@ TEST_P(StereoMatchBM, Accuracy)
     EXPECT_MAT_SIMILAR(disp_gold, disp, 1e-3);
 }
 
-INSTANTIATE_TEST_CASE_P(GPU_Calib3D, StereoMatchBM, testing::Combine(testing::Values(128),
+INSTANTIATE_TEST_CASE_P(OCL_Calib3D, StereoMatchBM, testing::Combine(testing::Values(128),
 	                                   testing::Values(19)));
 
+PARAM_TEST_CASE(StereoMatchBP, int, int, int, float, float, float, float)
+{
+    int ndisp_;
+    int iters_;
+    int levels_;
+    float max_data_term_;
+    float data_weight_;
+    float max_disc_term_;
+    float disc_single_jump_;
+    virtual void SetUp()
+    {
+        ndisp_          = GET_PARAM(0);
+        iters_          = GET_PARAM(1);
+        levels_         = GET_PARAM(2);
+        max_data_term_  = GET_PARAM(3);
+        data_weight_    = GET_PARAM(4);
+        max_disc_term_     = GET_PARAM(5);
+        disc_single_jump_  = GET_PARAM(6);
+    }
+};
+TEST_P(StereoMatchBP, Regression)
+{
+    Mat left_image  = readImage("stereobp/aloe-L.png");
+    Mat right_image = readImage("stereobp/aloe-R.png");
+    Mat disp_gold   = readImage("stereobp/aloe-disp.png", IMREAD_GRAYSCALE);
+    ocl::oclMat d_left, d_right;
+    ocl::oclMat d_disp;
+    Mat  disp;
+    ASSERT_FALSE(left_image.empty());
+    ASSERT_FALSE(right_image.empty());
+    ASSERT_FALSE(disp_gold.empty());
+    d_left.upload(left_image);
+    d_right.upload(right_image);
+    ocl::StereoBeliefPropagation bp(ndisp_, iters_, levels_, max_data_term_, data_weight_,
+        max_disc_term_, disc_single_jump_, CV_16S);
+    bp(d_left, d_right, d_disp);
+    d_disp.download(disp);
+    disp.convertTo(disp, disp_gold.depth());
+    EXPECT_MAT_NEAR(disp_gold, disp, 0.0, "");
+}
+INSTANTIATE_TEST_CASE_P(OCL_Calib3D, StereoMatchBP, testing::Combine(testing::Values(64),
+    testing::Values(8),testing::Values(2),testing::Values(25.0f),
+    testing::Values(0.1f),testing::Values(15.0f),testing::Values(1.0f)));
 #endif // HAVE_OPENCL