From f132aba540f66a248fec66272a417adebbccf12c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=AE=B6=E6=89=8D=20=E7=8E=8B?= Date: Thu, 2 May 2019 14:35:40 +0800 Subject: [PATCH] =?UTF-8?q?=E5=8C=85=E8=A3=85=E6=88=90=E5=87=BD=E6=95=B0?= =?UTF-8?q?=EF=BC=8C=E5=87=91=E4=BB=A3=E7=A0=81=E9=87=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- MV.py | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 MV.py diff --git a/MV.py b/MV.py new file mode 100644 index 0000000..fcb4250 --- /dev/null +++ b/MV.py @@ -0,0 +1,34 @@ +import cv2 as cv +import numpy as np +cap = cv.VideoCapture("./vid/zootopia.mkv") +ret, frame1 = cap.read() +prvs = cv.cvtColor(frame1,cv.COLOR_BGR2GRAY) +hsv = np.zeros_like(frame1) +hsv[...,1] = 255 +print("read") +cap.set(cv.CAP_PROP_POS_FRAMES,59760) +print("read done") +while(1): + for i in range(1,12): + cap.read() + ret, frame2 = cap.read() + next = cv.cvtColor(frame2,cv.COLOR_BGR2GRAY) + flow = cv.calcOpticalFlowFarneback(prvs,next, None, 0.5, 3, 15, 3, 5, 1.2, 0) + mag, ang = cv.cartToPolar(flow[...,0], flow[...,1]) + hsv[...,0] = ang*180/np.pi/2 + hsv[...,2] = cv.normalize(mag,None,0,255,cv.NORM_MINMAX) + bgr = cv.cvtColor(hsv,cv.COLOR_HSV2BGR) + cv.namedWindow("frame2", cv.WINDOW_NORMAL) + cv.namedWindow("frame", cv.WINDOW_NORMAL) + cv.imshow('frame2',bgr) + cv.imshow("frame",frame2) + cv.waitKey(1) + #k = cv.waitKey(0.1) & 0xff + #if k == 27: + # break + #elif k == ord('s'): + # cv.imwrite('opticalfb.png',frame2) + # cv.imwrite('opticalhsv.png',bgr) + prvs = next +cap.release() +cv.destroyAllWindows() \ No newline at end of file