pip install simple-filters
This is a collection of simple filters and trackers based on NumPy and SciPy, which are optimized for real-time performance. This is an example that applies tracking to non-recurrent methods like SMOKE with the DummyFilterStrategy:
A filter acts as a container for a time-series. The length of the time-series is kept constant after initial filling, according to the history_size
specified.
Currently, two filters are implemented:
- NumpyFilterStrategy: Applies a numpy function (e.g. numpy.mean) to the time-series.
- PolynomialFilterStrategy: Returns the filtered last item (and optionally predicts the next item) of a multi-dimensional time series using a polynomial regression. The strategy can be applied to sensor data to retain smoothness while ensuring low latency and avoiding offsets with outliers.
- DummyFilterStrategy: Simply returns the last item of the time-series.
Set up your filter:
from simple_filters import Filter, PolynomialFilterStrategy, NumpyFilterStrategy, DummyFilterStrategy
strategy_1 = PolynomialFilterStrategy(poly_degree=3, outlier_rejection_ratio=2.0)
strategy_2 = NumpyFilterStrategy(np.mean)
strategy_3 = DummyFilterStrategy()
filter = Filter(strategy_1, history_size=10)
Fill the history with a 1D array:
filter.update([1.0, 2.0])
filter.update([1.1, 2.1])
Get the last filtered item at the current_time, or by specifying a time step in the future (time=1
). Note that only PolynomialFilterStrategy allows for predicting the future.
result_current = filter.eval()
result_future = filter.eval(time=1)
Oftentimes, multiple objects must be tracked that also require filtering. SimpleFilters implements a simple multi-object tracker for this purpose. The tracker associates objects by applying minimum weight matching to a distance graph.
The following properties can be defined:
- distance_threshold: Maximum distance to match objects - when the threshold is exceeded, a new object will be created
- max_time_to_live: If an object is not seen, it is still retained for the given number of state updates
- time_to_birth: The number of observations needed until an object is born
- filter_prototype: A filter that will be cloned for each new appearing object
- distance_function: A lambda (x1, x2) that returns a distance between the two arrays
The PolynomialFilterStrategy is especially suitable for tracking, as it can predict the future state of the object according to its reconstructed polynomial:
from simple_filters import Filter, PolynomialFilterStrategy, Tracker
strategy = PolynomialFilterStrategy(poly_degree=3, outlier_rejection_ratio=2.0)
filter_prototype = Filter(strategy, history_size=10)
tracker = Tracker(filter_prototype, distance_threshold=1.0, time_to_live=1)
Update your tracker with a 2D array:
tracker.update([[1.0, 1.0], [2.0, 2.1]])
tracker.update([[1.1, 1.1], [1.9, 2.0], [3.1, 3.0]])
Retrieve the results:
# Access the list of objects like any filter:
list_of_objects = tracker.get_tracked_objects()
list_of_objects[0].eval(0) # get the latest state ([1.1, 1.1])
list_of_objects[0].id # get the tracking ID
# Or convert to a NumPy array:
np_array = tracker.to_numpy_array()
Simply run pytest
in the project directory.