trying (and failing?) to get hw acceleration

main
Shoofle 3 months ago
parent 0d43c980af
commit 7d2ff5d7e3
  1. 1
      Pipfile
  2. 11
      Pipfile.lock
  3. 137
      webcam.py

@ -5,6 +5,7 @@ name = "pypi"
[packages] [packages]
opencv-python = "*" opencv-python = "*"
wakepy = "*"
[dev-packages] [dev-packages]

11
Pipfile.lock generated

@ -1,7 +1,7 @@
{ {
"_meta": { "_meta": {
"hash": { "hash": {
"sha256": "9cd65918174126dae7310623cabe06ec6e5ce3f1bb0e255262749a78a55d3839" "sha256": "943975627c31fb661a3a1e3a77e80b2e5634baf439d615ccba41a14d7edf09b1"
}, },
"pipfile-spec": 6, "pipfile-spec": 6,
"requires": { "requires": {
@ -80,6 +80,15 @@
"index": "pypi", "index": "pypi",
"markers": "python_version >= '3.6'", "markers": "python_version >= '3.6'",
"version": "==4.10.0.84" "version": "==4.10.0.84"
},
"wakepy": {
"hashes": [
"sha256:4e7e820512e0e69f9aa5fdded2d3225e6f7c9f000814300e8c37d1fa88fe1664",
"sha256:fee0722b554a859724f6f47ec28db1d513f15097f235b09ebb1b58f545f90db2"
],
"index": "pypi",
"markers": "python_version >= '3.7'",
"version": "==0.9.1"
} }
}, },
"develop": {} "develop": {}

@ -1,86 +1,107 @@
from datetime import datetime, date, time, timedelta from datetime import datetime, date, time, timedelta
import os import os
import cv2 import cv2
from wakepy import keep
def five_minute_segment_from(the_time): def five_minute_segment_from(the_time):
return the_time.replace(minute=(the_time.minute // 1)*1, second=0, microsecond=0) # this is mocked out too be a one minute segment instead
return the_time.replace(second=0, microsecond=0)
#return the_time.replace(minute=(the_time.minute // 1)*1, second=0, microsecond=0)
def path_to_video_at(time_stamp): def path_to_video_at(time_stamp):
return 'videos/' + str(time_stamp) + '.mkv' return 'videos/' + str(time_stamp) + '.mkv' # mkv required for h264 encoding
#delay for long delay os.environ["OPENCV_FFMPEG_CAPTURE_OPTIONS"]="hwaccel;vaapi;hw_decoders_any;vaapi,vdpau" # maybe this enables hardware accelerated ffmpeg?
long_delay = timedelta(hours=1)
#delay for long delay
long_delay = timedelta(minutes=1)
queue = [(False, None)] * 100 #queue of recent frames
last_time_stamp = five_minute_segment_from(datetime.now()) last_time_stamp = five_minute_segment_from(datetime.now())
# live webcam feed # live webcam feed
camera = cv2.VideoCapture(0) camera = cv2.VideoCapture(0, cv2.CAP_FFMPEG)
if not camera.isOpened(): if not camera.isOpened():
print("Cannot open camera") print("Cannot open camera")
exit() exit()
# video writer # video writer
fourcc = cv2.VideoWriter_fourcc(*'H264') fourcc = cv2.VideoWriter_fourcc(*'H264')
out = cv2.VideoWriter(path_to_video_at(last_time_stamp), fourcc, camera.get(cv2.CAP_PROP_FPS), (int(camera.get(3)),int(camera.get(4)))) video_writer = cv2.VideoWriter(path_to_video_at(last_time_stamp),
fourcc,
camera.get(cv2.CAP_PROP_FPS),
(int(camera.get(3)),int(camera.get(4))),
(cv2.VIDEOWRITER_PROP_HW_ACCELERATION, cv2.VIDEO_ACCELERATION_ANY)
)
# video feed from previously captured video # video feed from previously captured video
old_time = cv2.VideoCapture(path_to_video_at(last_time_stamp)) old_time = cv2.VideoCapture(path_to_video_at(last_time_stamp), cv2.CAP_FFMPEG)
# fullscreen window yess good
cv2.namedWindow("mirror", cv2.WINDOW_NORMAL) cv2.namedWindow("mirror", cv2.WINDOW_NORMAL)
cv2.setWindowProperty("mirror", cv2.WND_PROP_FULLSCREEN, cv2.WINDOW_FULLSCREEN) cv2.setWindowProperty("mirror", cv2.WND_PROP_FULLSCREEN, cv2.WINDOW_FULLSCREEN)
idx = 0 queue = [(False, None)] * 100 #queue of recent frames
offset = 0 idx = 0 # current index into the frame queue
longterm = False short_term_delay_frames = 20 # hoow many frames back are we going in the buffer when we're on a short term delay?
while(True): longterm = False # are we shoowing a long term delay? (versus a sshort term delay)
if last_time_stamp != five_minute_segment_from(datetime.now()):
# find file older than our long delay with keep.presenting():
old_file = path_to_video_at(five_minute_segment_from(datetime.now() - long_delay - timedelta(minutes=5))) while(True):
print(f"deleting old file {old_file}") if last_time_stamp != five_minute_segment_from(datetime.now()):
os.remove(old_file) # find file older than our long delay
old_file = path_to_video_at(five_minute_segment_from(datetime.now() - long_delay - timedelta(minutes=1)))
out.release() if os.path.isfile(old_file):
out = cv2.VideoWriter(path_to_video_at(last_time_stamp), print(f"deleting old file {old_file}")
fourcc, os.remove(old_file)
camera.get(cv2.CAP_PROP_FPS),
(int(camera.get(3)), int(camera.get(4))) video_writer.release()
) newly_recording_file = path_to_video_at(five_minute_segment_from(datetime.now()))
if old_time.isOpened():
old_time.release() print(f"opening {newly_recording_file} too record this upcoming ssegment to")
video_writer = cv2.VideoWriter(newly_recording_file,
new_file = path_to_video_at(five_minute_segment_from(datetime.now() - timedelta(minutes=10))) fourcc,
pint(f"opening the previously recorded file {new_file} to read from for long mirror") camera.get(cv2.CAP_PROP_FPS),
old_time = cv2.VideoCapture(new_file) (int(camera.get(3)), int(camera.get(4))),
(cv2.VIDEOWRITER_PROP_HW_ACCELERATION, cv2.VIDEO_ACCELERATION_ANY)
last_time_stamp = five_minute_segment_from(datetime.now()) )
ret, frame = camera.read()
frame = cv2.flip(frame, 1) if old_time.isOpened():
old_time.release()
# record the frame to the appropriate file, and also to the short term buffer
out.write(frame) new_file = path_to_video_at(five_minute_segment_from(datetime.now() - long_delay))
queue[idx] = ret, frame print(f"opening the previously recorded file {new_file} to read from for long mirror")
old_time = cv2.VideoCapture(new_file, cv2.CAP_FFMPEG)
prev_frame_valid, prev_frame = queue[(idx - offset) % len(queue)]
old_frame_valid, old_frame = old_time.read()
if not longterm and prev_frame_valid: last_time_stamp = five_minute_segment_from(datetime.now())
cv2.imshow('mirror', prev_frame)
elif longterm and old_frame_valid: ret, frame = camera.read()
cv2.imshow('mirror', old_frame) frame = cv2.flip(frame, 1)
key = cv2.waitKey(16) # listen for keyprress for some number of ms # record the frame to the appropriate file, and also to the short term buffer
if key & 0xFF == ord('q') or key == 27: # q or esc video_writer.write(frame)
break queue[idx] = ret, frame
if key & 0xFF == ord('a'):
offset += 1 prev_frame_valid, prev_frame = queue[(idx - short_term_delay_frames) % len(queue)]
if key & 0xFF == ord('d'): old_frame_valid, old_frame = old_time.read()
offset -= 1 if not longterm and prev_frame_valid:
if key & 0xFF == ord('z'): cv2.imshow('mirror', prev_frame)
longterm = not longterm elif longterm and old_frame_valid:
cv2.imshow('mirror', old_frame)
idx = (idx + 1) % len(queue)
key = cv2.waitKey(16) # listen for keyprress for 16 ms
if key & 0xFF == ord('q') or key == 27: # q or esc
break
if key & 0xFF == ord('a'):
short_term_delay_frames += 1
if key & 0xFF == ord('d'):
short_term_delay_frames -= 1
if key & 0xFF == ord('z'):
longterm = not longterm
idx = (idx + 1) % len(queue)
# After the loop release the cap object # After the loop release the cap object
camera.release() camera.release()

Loading…
Cancel
Save