[英]How to calculate Optical Flow magnitude in OpenCV
我正在嘗試查看給定的兩個視頻幀有多大差異。 我的目標是計算一個單一值,以顯示這些幀內的對象移動的速度。
我可以在下面計算HSV和幅度矩陣的光流矩陣。 但是我不知道如何計算平均總運動幅度。 如何從這些矩陣中計算出來?
def optical_flow(one, two):
one_g = cv2.cvtColor(one, cv2.COLOR_RGB2GRAY)
two_g = cv2.cvtColor(two, cv2.COLOR_RGB2GRAY)
hsv = np.zeros((120, 320, 3))
# set saturation
hsv[:,:,1] = cv2.cvtColor(two, cv2.COLOR_RGB2HSV)[:,:,1]
# obtain dense optical flow paramters
flow = cv2.calcOpticalFlowFarneback(one_g, two_g, flow=None,
pyr_scale=0.5, levels=1, winsize=15,
iterations=2,
poly_n=5, poly_sigma=1.1, flags=0)
# convert from cartesian to polar
mag, ang = cv2.cartToPolar(flow[..., 0], flow[..., 1])
# hue corresponds to direction
hsv[:,:,0] = ang * (180/ np.pi / 2)
# value corresponds to magnitude
hsv[:,:,2] = cv2.normalize(mag,None,0,255,cv2.NORM_MINMAX)
# convert HSV to int32's
hsv = np.asarray(hsv, dtype= np.float32)
rgb_flow = cv2.cvtColor(hsv,cv2.COLOR_HSV2RGB)
return rgb_flow
rgb_flow
是一個3D數組,如下所示:
[[[0 0 0]
[0 0 0]
[0 0 0]
...
[0 0 0]
[0 0 0]
[0 0 0]]
[[0 0 0]
[0 0 0]
[0 0 0]
...
[0 0 0]
[0 0 0]
[0 0 0]]
...
[[0 0 0]
[0 0 0]
[0 0 0]
...
[0 0 0]
[0 0 0]
[0 0 0]]]
mag
矩陣是2D數組,如下所示:
[[3.2825139e-03 3.9561605e-03 4.8938910e-03 ... 3.7310597e-02
3.2986153e-02 2.5520157e-02]
[4.9569397e-03 6.3276174e-03 7.7017904e-03 ... 3.9564677e-02
3.2582227e-02 2.6329078e-02]
...
[6.9548332e-06 8.3683852e-05 6.0906638e-03 ... 8.3484064e-04
6.4721738e-04 2.9505073e-04]]
import numpy as np
np.mean(mag)
這將為您提供平均幅度。 即使numpy陣列mag是2d,默認情況下np.mean也會將其減少為一個數字。 此外,由於均值對異常值敏感,因此中位數可能會更好地為您服務。
np.median(mag)
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.