| Branch | Build status |
|---|---|
master |
 |
develop |
 |
Python and C implementations of an ellipse fitting algorithm in double and fixed point precision.
The implementations here were never meant for production, but were written as an examination of how much speedup could be attained from implementing ellipse fitting methods in fixed point and at what precision cost.
The ellipse fitting method in [2] (an improved version of the one in [1]) was implemented.
Install simply by calling:
pip install git+https://www.github.com/hbldh/b2ac
Numpy needs to be installed prior to b2ac, since it compiles the C extension using Numpy headers.
Test with nosetests:
nosetests tests
An ellipse can be fitted using Python methods:
import numpy as np
import b2ac.preprocess
import b2ac.fit
import b2ac.conversion
points = np.array([[3, 0], [0, 5], [3, 10], [6, 5]])
# Optional removal of mean value from points to obtain better
# fitting performance, especially in integer precision.
points, x_mean, y_mean = b2ac.preprocess.remove_mean_values(points)
# Fit using NumPy methods in double precision.
conic_numpy = b2ac.fit.fit_improved_B2AC_numpy(points)
# Fit using own written methods in double precision.
conic_double = b2ac.fit.fit_improved_B2AC_double(points)
# Fit using own written methods in 64-bit integer precision.
conic_int = b2ac.fit.fit_improved_B2AC_int(points)
# Convert from conic coefficient form to general ellipse form.
general_form_numpy = b2ac.conversion.conic_to_general_1(conic_numpy)
general_form_numpy[0][0] += x_mean
general_form_numpy[0][1] += y_mean
general_form_double = b2ac.conversion.conic_to_general_1(conic_double)
general_form_double[0][0] += x_mean
general_form_double[0][1] += y_mean
general_form_int = b2ac.conversion.conic_to_general_int(conic_int)
general_form_int[0][0] += x_mean
general_form_int[0][1] += y_meanThe mathematical notation described below follows the one used in [2].
First, the scatter matrix S is calculated according to regular specifications, and the M matrix as well. The inverse of S3 is calculated through definition of 3x3 matrix inverse.
A QR algorithm with largest value shift, using Givens rotations for both tridiagonalization and actual QR steps, (See [3]) is then used to find eigenvalues of the matrix M. With these eigenvalues, we apply an inverse iteration method for calculating the corresponding eigenvectors.
The algorithms returns the conic coefficients defining one fitted ellipse. These can then be transformed to general form: center point, axes lengths and rotation angle.
The integer version uses 64 bit integers!
The calculations of S and M has special handling. The fact that this method uses a lot of squared values can easily lead to overflow, especially if 32 bit integers are used.
The major thing to remember here is that when calculating inverses, the division of the determinant is postponed, thus using the determinant as a scale factor during calculations. This scale factor can then be removed first when the sought eigenvector has been found.
For integer versions, it is of great importance to first remove the mean values from the points, thus making all values much smaller. This will drastically improve the fitting results.
Another important aspect is the number of points used for estimating an
ellipse. Using >100 points, the matrix values in the estimation also grows
and the possibility of overflow is increased. Subsampling of points is recommended.
-
[Fitzgibbon, A., Pilu, M., & Fisher, R.B., "Direct least square fitting of ellipses," Pattern Analysis and Machine Intelligence, IEEE Transactions on , vol.21, no.5, pp.476-480, May 1999] (http://research.microsoft.com/pubs/67845/ellipse-pami.pdf)
-
Golub, G.H. & Van Loan, C.F., 2012. Matrix Computations (4th Ed.). Johns Hopkins University Press, Baltimore, MD, USA.