Merge pull request #3369 from sturkmen72:update_find_ellipses

Update find ellipses

Author: asmorkalov

modules/ximgproc/include/opencv2/ximgproc/find_ellipses.hpp

@@ -17,8 +17,8 @@ namespace ximgproc {
 @brief Finds ellipses fastly in an image using projective invariant pruning.
 *
 * The function detects ellipses in images using projective invariant pruning.
-* For more details about this implementation, please see
-* [JIA2017FAST]  Jia, Qi et al, (2017).
+* For more details about this implementation, please see @cite jia2017fast
+* Jia, Qi et al, (2017).
 * A Fast Ellipse Detector using Projective Invariant Pruning. IEEE Transactions on Image Processing.
 *
 @param image input image, could be gray or color.

modules/ximgproc/samples/find_ellipses.py

@@ -0,0 +1,46 @@
+#!/usr/bin/python
+
+'''
+This example illustrates how to use cv.ximgproc.findEllipses function.
+
+Usage:
+    find_ellipses.py [<image_name>]
+    image argument defaults to stuff.jpg
+'''
+
+# Python 2/3 compatibility
+from __future__ import print_function
+
+import numpy as np
+import cv2 as cv
+import sys
+import math
+
+def main():
+    try:
+        fn = sys.argv[1]
+    except IndexError:
+        fn = 'stuff.jpg'
+
+    src = cv.imread(cv.samples.findFile(fn))
+    cv.imshow("source", src)
+
+    ells = cv.ximgproc.findEllipses(src,scoreThreshold = 0.4, reliabilityThreshold = 0.7, centerDistanceThreshold = 0.02)
+
+    if ells is not None:
+        for i in range(len(ells)):
+            center = (int(ells[i][0][0]), int(ells[i][0][1]))
+            axes = (int(ells[i][0][2]),int(ells[i][0][3]))
+            angle = ells[i][0][5] * 180 / math.pi
+            color = (0, 0, 255)
+            cv.ellipse(src, center, axes, angle,0, 360, color, 2, cv.LINE_AA)
+
+    cv.imshow("detected ellipses", src)
+    cv.waitKey(0)
+    print('Done')
+
+
+if __name__ == '__main__':
+    print(__doc__)
+    main()
+    cv.destroyAllWindows()

modules/ximgproc/src/find_ellipses.cpp

@@ -1974,15 +1974,14 @@ void findEllipses(
     edi.detect(grayImage, ellipseResults);
 
     // convert - ellipse format to std::vector<Vec6f>
-    auto ellipseSize = unsigned(ellipseResults.size());
-    Mat _ellipses(1, ellipseSize, CV_32FC(6));
-    for (unsigned i = 0; i < ellipseSize; i++) {
+    std::vector<Vec6f> _ellipses;
+    for (size_t i = 0; i < ellipseResults.size(); i++) {
         Ellipse tmpEll = ellipseResults[i];
         Vec6f tmpVec(tmpEll.center.x, tmpEll.center.y, tmpEll.a, tmpEll.b, tmpEll.score,
                      tmpEll.radius);
-        _ellipses.at<Vec6f>(i) = tmpVec;
+        _ellipses.push_back(tmpVec);
     }
-    _ellipses.copyTo(ellipses);
+    Mat(_ellipses).copyTo(ellipses);
 }
 } // namespace ximgproc
 } // namespace cv