wangjiacai/StereoVidComfort
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Almost done.

Browse Source
This commit is contained in:
家才 王 2019-05-15 14:19:34 +08:00
parent d978f3e818
commit b5917875e1
2 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
MV.py
View File

@ -8,7 +8,7 @@ prvs = cv.cvtColor(frame1, cv.COLOR_BGR2GRAY)
hsv = np.zeros_like(frame1) hsv = np.zeros_like(frame1)
hsv[..., 1] = 255 hsv[..., 1] = 255
print("jump ahead") print("jump ahead")
cap.set(cv.CAP_PROP_POS_FRAMES, 12000) cap.set(cv.CAP_PROP_POS_FRAMES, 4)
print("jump done") print("jump done")
while(1): while(1):

19
StereoVidComfort.py
View File

@ -113,7 +113,7 @@ if __name__ == "__main__":
next = imgR # 当前帧的右画面，用于运动矢量计算 next = imgR # 当前帧的右画面，用于运动矢量计算
hsv = getMotionVector(prvs, next) hsv = getMotionVector(prvs, next)
hsv_bak = hsv.copy()
# 计算深度图,disparity越大，景深越小，物体越近 # 计算深度图,disparity越大，景深越小，物体越近
disparity = getDepthMap(imgL, imgR) disparity = getDepthMap(imgL, imgR)
@ -204,18 +204,21 @@ if __name__ == "__main__":
# cv2.imshow('imgR', imgR) # cv2.imshow('imgR', imgR)
# 显示当前帧的运动矢量的hsv表示 # 显示当前帧的运动矢量的hsv表示
# bgr = cv2.cvtColor(hsv, cv2.COLOR_HSV2BGR) # hsv转为rgb用于显示 bgr = cv2.cvtColor(hsv_bak, cv2.COLOR_HSV2BGR) # hsv转为rgb用于显示
# cv2.namedWindow("MotionVector", cv2.WINDOW_NORMAL) cv2.namedWindow("MotionVector", cv2.WINDOW_NORMAL)
# cv2.imshow("MotionVector", bgr) cv2.imshow("MotionVector", bgr)
# cv2.waitKey(1) cv2.waitKey(1)
# 显示当前帧的景深图 # 显示当前帧的景深图
# plt.title("DepthMap") plt.title("DepthMap")
# plt.imshow(disparity) plt.imshow(disparity)
# plt.pause(0.1)
# input("press Enter to continue")
# 运动矢量的直方图，方便查看数值 # 运动矢量的直方图，方便查看数值
# plt.title("MotionVector") # plt.title("MotionVector")
# plt.imshow(hsv[...,2]) # plt.imshow(hsv[...,2])
# plt.show() # plt.show()
# plt.pause(0.1) plt.pause(0.1)
input("press Enter to continue") input("press Enter to continue")
prvs = next # 当前帧覆盖上一帧，继续计算 prvs = next # 当前帧覆盖上一帧，继续计算
print("TotalFrameCalculated: ", framesCalculated) print("TotalFrameCalculated: ", framesCalculated)