深度学习-实时视频处理(FPS计数低)

时间:2018-12-19 05:17:53

标签: python-3.x opencv keras deep-learning video-streaming

@developers,需要有关实时视频处理的帮助。我对FPS有问题。 (结果视频不流畅)。

我想到的解决方案是:  1.拆分代码并运行多个线程。  2.提高Fps速率

我附上代码,请看一下,为我找到一个更好的解决方案。

from matplotlib import pyplot as plt
import cv2
import numpy as np
import sys, os

sys.path.append('/home/developer/Desktop/sss/keras-deeplab-v3-plus- master/')
from model import Deeplabv3
sys.path.append('/home/developer/Desktop/sss/imutils-master')
from imutils.video import WebcamVideoStream

deeplab_model = Deeplabv3()
vid = WebcamVideoStream(src=0).start()
cv2.namedWindow("result", cv2.WINDOW_NORMAL)
blurValue = (41,41)


while True:
   frame = vid.read()
   if frame is None:
      break
   w, h, _ = frame.shape
   ratio = 512. / np.max([w,h])

   resized = cv2.resize(frame,(int(ratio*h),int(ratio*w)))
   resized = resized / 127.5 - 1.
   pad_x = int(512 - resized.shape[0])
   resized2 = np.pad(resized,((0,pad_x),(0,0),(0,0)),mode='constant')    
   res = deeplab_model.predict(np.expand_dims(resized2,0))
   labels = np.argmax(res.squeeze(),-1)    

   labels = labels[:-pad_x-25]
   mask = labels == 0    
   resizedFrame = cv2.resize(frame, (labels.shape[1],labels.shape[0]))
   blur = cv2.GaussianBlur(resizedFrame,blurValue,0)  
   resizedFrame[mask] = blur[mask]
   cv2.imshow("result",resizedFrame)
   if cv2.waitKey(1) & 0xFF == ord('q'):
      break
vid.stop()
cv2.destroyAllWindows()

请帮助我顺利过渡结果视频。

预先感谢

0 个答案:

没有答案