LucasKanade.py

#!/usr/bin/env python
"""
This function isn't working yet.
"""

import imageio
from scipy.ndimage.filters import gaussian_filter
from argparse import ArgumentParser
from pyoptflow import LucasKanade, getPOI, gaussianWeight
from pyoptflow import getimgfiles
from pyoptflow.plots import compareGraphsLK


def main():
    p = ArgumentParser(description="Pure Python Lucas Kanade Optical Flow")
    p.add_argument("stem", help="path/stem of files to analyze")
    p.add_argument("pat", help="pattern glob")
    p = p.parse_args()

    lucas_kanade(p.stem, p.pat)


def lucas_kanade(stem, pat: str, kernel: int = 5, Nfilter: int = 7):
    flist = getimgfiles(stem, pat)

    # %% priming read
    im1 = imageio.imread(flist[0], as_gray=True)

    # %% evaluate the first frame's POI
    X = im1.shape[1] // 16
    Y = im1.shape[0] // 16
    poi = getPOI(X, Y, kernel)
    # % get the weights
    W = gaussianWeight(kernel)
    # %% loop over all images in directory
    for i in range(1, len(flist)):
        im2 = imageio.imread(flist[i], as_gray=True)

        im2 = gaussian_filter(im2, Nfilter)

        V = LucasKanade(im1, im2, kernel, poi, W)

        compareGraphsLK(im2, poi, V)

        im1 = im2.copy()  # type: ignore[attr-defined]


if __name__ == "__main__":
    main()