def demo(stem):
flist = getimgfiles(stem)
ext = flist[0].suffix
for i in range(len(flist)-1):
fn1 = f'{stem}.{i}{ext}'
im1 = imread(fn1,flatten=True).astype(float) #flatten=True is rgb2gray
# Iold = gaussian_filter(Iold,FILTER)
fn2 = f'{stem}.{i+1}{ext}'
im2 = imread(fn2,flatten=True).astype(float)
# Inew = gaussian_filter(Inew,FILTER)
U,V = HornSchunck(im1, im2, 1., 100)
compareGraphs(U,V, im2)
return U,V
def horn_schunck(stem, pat:str):
flist = getimgfiles(stem, pat)
for i in range(len(flist)-1):
fn1 = flist[i]
im1 = imageio.imread(fn1, as_gray=True)
# Iold = gaussian_filter(Iold,FILTER)
fn2 = flist[i+1]
im2 = imageio.imread(fn2, as_gray=True)
# Inew = gaussian_filter(Inew,FILTER)
U,V = HornSchunck(im1, im2, 1., 100)
compareGraphs(U,V, im2, fn=fn2.name)
return U,V
def horn_schunck(stem: Path, pat: str, alpha: float, Niter: int,
verbose: bool):
flist = getimgfiles(stem, pat)
for i in range(len(flist) - 1):
fn1 = flist[i]
im1 = imageio.imread(fn1, as_gray=True)
# Iold = gaussian_filter(Iold,FILTER)
fn2 = flist[i + 1]
im2 = imageio.imread(fn2, as_gray=True)
# Inew = gaussian_filter(Inew,FILTER)
U, V = HornSchunck(im1, im2, alpha=1.0, Niter=100)
compareGraphs(U, V, im2, fn=fn2.name)
return U, V