From d96ee5ee607208622837023273b4ee8c7b509216 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=AE=B6=E6=89=8D=20=E7=8E=8B?= <jiacai_wang@qq.com>
Date: Mon, 29 Apr 2019 18:58:03 +0800
Subject: [PATCH] =?UTF-8?q?plt.show=E9=98=BB=E5=A1=9E=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 StereoVidComfort.py | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/StereoVidComfort.py b/StereoVidComfort.py
index 7e2e5c2..659761f 100644
--- a/StereoVidComfort.py
+++ b/StereoVidComfort.py
@@ -10,19 +10,20 @@ cap = cv2.VideoCapture('./vid/zootopia.mkv')
 frameCount = cap.get(cv2.CAP_PROP_FRAME_COUNT)
 frameRate = cap.get(cv2.CAP_PROP_FPS)
 
-for frameID in range(int(frameRate), int(frameCount), int(frameRate/5)):
+for frameID in range(int(frameRate), int(frameCount), int(frameRate*100)):
     cap.set(cv2.CAP_PROP_POS_FRAMES, frameID)
     isSuccess, img = cap.read()
     if isSuccess:
+        cv2.namedWindow("img",cv2.WINDOW_NORMAL);
         cv2.imshow('img', img)
         imgL = np.split(img, 2, 1)[0]
         imgR = np.split(img, 2, 1)[1]
         cv2.waitKey(1)
         stereo = cv2.StereoSGBM_create(numDisparities=96, blockSize=7)
         disparity = stereo.compute(imgL, imgR)
-        plt.title("SGBM")
+        plt.title("DepthMap")
         plt.imshow(disparity)
-        plt.show()
+        plt.pause(0.1)
 
 
 
@@ -76,3 +77,14 @@ for frameID in range(int(frameRate), int(frameCount), int(frameRate/5)):
 
 #    plt.imshow(disparity)
 #    plt.show()
+# import numpy as np
+# import cv2
+# from matplotlib import pyplot as plt
+# 
+# imgL = cv2.imread('tsukuba_l.png',0)
+# imgR = cv2.imread('tsukuba_r.png',0)
+# 
+# stereo = cv2.StereoBM_create(numDisparities=16, blockSize=15)
+# disparity = stereo.compute(imgL,imgR)
+# plt.imshow(disparity,'gray')
+# plt.show()