cuda_python/videoProcessing.py at master · Alpha-Coderz/cuda_python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
from AverageFilter import *
import numpy as np
import cv2
import time

cap = cv2.VideoCapture(0)

capture_fps= cap.get(cv2.CAP_PROP_FPS)

if (cap.isOpened()== False):
  print("Error opening video stream or file")

while(cap.isOpened()):
  start = time.time()

  ret, frame = cap.read()
  if ret == True:
    x,y,z=frame.shape
    #splitting R,G and B channel
    frame_r=frame[:,:,2]
    frame_g=frame[:,:,1]
    frame_b=frame[:,:,0]
    d_img_r=cuda.to_device(np.ascontiguousarray(frame_r))
    d_img_g=cuda.to_device(np.ascontiguousarray(frame_g))
    d_img_b=cuda.to_device(np.ascontiguousarray(frame_b))
    TPB=(kernel_width,kernel_width)
    BPG=(x,y)
  	#operating on R,G and B channel
    avg_filter[BPG,TPB](d_img_r,x-1,y-1)
    avg_filter[BPG,TPB](d_img_g,x-1,y-1)
    avg_filter[BPG,TPB](d_img_b,x-1,y-1)
    #combining three channels
    frame_smooth=np.dstack((d_img_g.copy_to_host(),d_img_g.copy_to_host(),d_img_r.copy_to_host()))


    cv2.imshow('Smoothed Image',frame_smooth)
    cv2.imshow('Original',frame)

    if cv2.waitKey(25) & 0xFF == ord('q'):
      break


  else:
    break
  end=time.time()-start
  fps=60/end
  print(fps/capture_fps)


cap.release()


cv2.destroyAllWindows()