Untitled
unknown
plain_text
2 years ago
849 B
5
Indexable
import numpy as np import matplotlib.pyplot as plt from ex1_utils import rotate_image, show_flow from of_methods import lucas_kanade, horn_schunck im1 = np.random.rand(200, 200).astype(np.float32) im2 = im1.copy() im2 = rotate_image(im2, -1) U_lk, V_lk = lucas_kanade(im1, im2, 3) U_hs, V_hs = horn_schunck(im1, im2, 10000, 0.5) fig1, ((ax1_11, ax1_12), (ax1_21, ax1_22)) = plt.subplots(2, 2) ax1_11.imshow(im1) ax1_12.imshow(im2) show_flow(U_lk, V_lk, ax1_21, type='angle') show_flow(U_lk, V_lk, ax1_22, type='field', set_aspect=True) fig1.suptitle('Lucas−Kanade Optical Flow') fig2, ((ax2_11, ax2_12), (ax2_21, ax2_22)) = plt.subplots(2, 2) ax2_11.imshow(im1) ax2_12.imshow(im2) show_flow(U_hs, V_hs, ax2_21, type='angle') show_flow(U_hs, V_hs, ax2_22, type='field', set_aspect=True) fig2.suptitle('Horn−Schunck Optical Flow') plt.show()
Editor is loading...