aruco::Board::create: fix debug assertion fail

8 years ago · 820af3a03a
parent 933f53e47f
commit 820af3a03a
1 changed files with 5 additions and 8 deletions
--- a/modules/aruco/src/aruco.cpp
+++ b/modules/aruco/src/aruco.cpp
@ -1284,14 +1284,11 @@ Ptr<Board> Board::create(InputArrayOfArrays objPoints, const Ptr<Dictionary> &di
    for (unsigned int i = 0; i < objPoints.total(); i++) {
        std::vector<Point3f> corners;
        Mat corners_mat = objPoints.getMat(i);
-        if(objPoints.type() == CV_32FC1)
-        {
-            CV_Assert(corners_mat.total() == 12);
-        }
-        else
-        {
-            CV_Assert(corners_mat.total() == 4);
-        }
+
+        if(corners_mat.type() == CV_32FC1)
+            corners_mat = corners_mat.reshape(3);
+        CV_Assert(corners_mat.total() == 4);
+
        for (int j = 0; j < 4; j++) {
            corners.push_back(corners_mat.at<Point3f>(j));
        }