-
Notifications
You must be signed in to change notification settings - Fork 0
/
Play_the_movie_solid_yellow_left.py
50 lines (40 loc) · 1.31 KB
/
Play_the_movie_solid_yellow_left.py
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
import cv2
from PIL import ImageGrab
import time
from numpy import ones,vstack
from numpy.linalg import lstsq
import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import numpy as np
import cv2
import math
from PIL import ImageGrab
import time
from numpy import ones,vstack
from numpy.linalg import lstsq
#from . import 02_udacity_photo_line_detection_test
kernel_size = 5
low_threshold = 50
high_threshold = 150
def origin_video():
cap = cv2.VideoCapture('test_videos/solidYellowLeft.mp4')
frame_counter = 0
while(True):
# Capture frame-by-frame
ret, frame = cap.read()
frame_counter += 1
#If the last frame is reached, reset the capture and the frame_counter
if frame_counter == cap.get(cv2.CAP_PROP_FRAME_COUNT):
frame_counter = 0 #Or whatever as long as it is the same as next line
cap.set(cv2.CAP_PROP_POS_FRAMES, 0)
# Our operations on the frame come here
RGB = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
# Display the resulting frame
cv2.namedWindow("solidYellowLeft", cv2.WINDOW_NORMAL)
cv2.resizeWindow("solidYellowLeft", 640,480)
cv2.imshow("solidYellowLeft",RGB)
if cv2.waitKey(27) & 0xFF == ord('q'):
break
cap.release()
cv2.destroyAllWindows()
origin_video()