-
Notifications
You must be signed in to change notification settings - Fork 11
/
save_video.py
143 lines (97 loc) · 3.85 KB
/
save_video.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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
#####################################################################
# Example : generic example for video processing from a video file
# specified as video.avi or from an
# attached web camera and saving to a video file
# Author : Toby Breckon, [email protected]
# Copyright (c) 2015 School of Engineering & Computing Science,
# Copyright (c) 2019 Dept Computer Science,
# Durham University, UK
# License : LGPL - http://www.gnu.org/licenses/lgpl.html
#####################################################################
import cv2
import argparse
import sys
#####################################################################
keep_processing = True
# parse command line arguments for camera ID or video file
parser = argparse.ArgumentParser(
description='Perform ' +
sys.argv[0] +
' example operation on incoming camera/video image')
parser.add_argument(
"-c",
"--camera_to_use",
type=int,
help="specify camera to use",
default=0)
parser.add_argument(
"-r",
"--rescale",
type=float,
help="rescale image by this factor",
default=1.0)
args = parser.parse_args()
video_width = 640
video_height = 480
#####################################################################
# define video capture object
try:
# to use a non-buffered camera stream (via a separate thread)
import camera_stream
cap = camera_stream.CameraVideoStream(use_tapi=False)
except BaseException:
# if not then just use OpenCV default
print("INFO: camera_stream class not found - camera input may be buffered")
cap = cv2.VideoCapture()
# define display window name
window_name = "Live Camera Input -> Video File" # window name
# define video writer (video: 640 x 480 @ 25 fps)
fourcc = cv2.VideoWriter_fourcc('M', 'J', 'P', 'G')
output = cv2.VideoWriter('output.avi', fourcc, 25.0,
(video_width, video_height))
# if command line arguments are provided try to read video_file
# otherwise default to capture from attached H/W camera
if ((cap.open("input.avi")) or (cap.open(args.camera_to_use))):
# create window by name (as resizable)
cv2.namedWindow(window_name, cv2.WINDOW_NORMAL)
while (keep_processing):
# if video file or camera successfully open then read frame from video
if (cap.isOpened):
ret, frame = cap.read()
# when we reach the end of the video (file) exit cleanly
if (ret == 0):
keep_processing = False
continue
# rescale if specified
if (args.rescale != 1.0):
frame = cv2.resize(
frame, (0, 0), fx=args.rescale, fy=args.rescale)
# *** do any processing here ****
# write the frame to file (first resizing)
frame2 = cv2.resize(
frame,
(video_width,
video_height),
interpolation=cv2.INTER_CUBIC)
output.write(frame2)
# display image
cv2.imshow(window_name, frame)
# start the event loop - essential
# cv2.waitKey() is a keyboard binding function (argument is the time in
# ms). It waits for specified milliseconds for any keyboard event.
# If you press any key in that time, the program continues.
# If 0 is passed, it waits indefinitely for a key stroke.
# (bitwise and with 0xFF to extract least significant byte of
# multi-byte response)
key = cv2.waitKey(1) & 0xFF # wait 1ms only
# e.g. if user presses "x" then exit
if (key == ord('x')):
keep_processing = False
# close all windows
cv2.destroyAllWindows()
# Release everything if job is finished
cap.release()
output.release()
else:
print("No video file specified or camera connected.")
#####################################################################