Deep Learning - обработка видео в реальном времени (низкое количество FPS) - PullRequest
Новая
       11

Deep Learning - обработка видео в реальном времени (низкое количество FPS)

0 голосов
/ 19 декабря 2018

@ разработчикам, нужна помощь в обработке видео в реальном времени.У меня есть проблема на FPS.(исходное видео не гладкое).

решение, о котором я думаю, это: 1. разделить код и запустить несколько потоков.2. увеличить частоту кадров в секунду

Я прилагаю код, пожалуйста, посмотрите и найдите мне лучшее решение. 

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()

Пожалуйста, помогите мне в плавном переходе к видео результата.

Заранее спасибо

...