From c20f695f4fcba432ddf972e58a91128589d5c7c8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E4=BB=BB=E5=B9=BF=E8=BE=89?= Date: Sat, 7 Apr 2018 17:40:48 +0800 Subject: [PATCH] viz flow by cv2 --- demo.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/demo.py b/demo.py index e1531b8..59cfdeb 100644 --- a/demo.py +++ b/demo.py @@ -4,7 +4,15 @@ import tensorflow as tf import scipy.io as sio from tvnet import TVNet - +def viz_flow(flow): + h, w = flow.shape[:2] + hsv = np.zeros((h, w, 3), np.uint8) + mag, ang = cv2.cartToPolar(flow[...,0], flow[...,1]) + hsv[...,0] = ang*180/np.pi/2 + hsv[...,1] = cv2.normalize(mag,None,0,255,cv2.NORM_MINMAX) + hsv[...,2] = 255 + bgr = cv2.cvtColor(hsv,cv2.COLOR_HSV2BGR) + return bgr flags = tf.app.flags flags.DEFINE_integer("scale", 5, " TVNet scale [3]") flags.DEFINE_integer("warp", 5, " TVNet warp [1]") @@ -51,3 +59,5 @@ os.mkdir('result') res_path = os.path.join('result', 'result.mat') sio.savemat(res_path, {'flow': flow_mat}) +flow_bgr = viz_flow(flow_mat) +cv2.imwrite(os.path.join('result', 'result.png'), flow_bgr)