drastically reduced latency

This commit is contained in:
Cowwan 2024-02-07 13:04:20 -05:00
parent cd06b323c1
commit 8236a8acd0
3 changed files with 22 additions and 22 deletions

View file

@ -112,27 +112,27 @@ frame_start = 0.0
peak_time_between = 0.0
peak_camera_latency = 0.0
total_camera_latency = 0.0
sleepTimer = 0.0
failures = 0
tracker = Tracker(width, height, featureType, threshold=args.threshold, silent=silent, model_type=args.model, detection_threshold=args.detection_threshold)
#don't start until the webcam is ready
#don't start until the webcam is ready, then give it a little more time to fill it's buffer
while frameQueue.qsize() < 1:
time.sleep(0.2)
time.sleep(0.1)
time.sleep(0.1)
face = None
totalFrameLatency = 0
try:
while True:
#clearing these so there's no stale data to confuse the checks
packet = None
frame_start = time.perf_counter()
frame_count += 1
#if the camera process doesn't have a frame for us we just skip everything
#in my testing the camera was consistently waiting on the main process
#so if it's *that far behind* it probably needs a frame or two to catch up
if frameQueue.qsize() > 0:
frame, camera_latency = frameQueue.get()
if True:
frame, camera_latency, totalFrameLatency = frameQueue.get()
frame_get = time.perf_counter()
#If I don't wait a few frames to start tracking I get wild peak frame times, like 500ms
if frame_count > 5:
peak_camera_latency = max(camera_latency, peak_camera_latency)
@ -164,15 +164,15 @@ try:
if visualizeFlag:
visualize(frame, face, previewFrameQueue, face_center, face_radius)
duration = time.perf_counter() - frame_start
duration = time.perf_counter() - frame_get
total_active_time += duration
peak_frame_time = max(peak_frame_time, duration)
if duration < target_duration:
time.sleep(target_duration - duration)
else:
#all caps because if this is consistently happening it's an issue
print("CANNOT MAINTAIN FRAMERATE!")
sleepTimer = max(time.perf_counter() - frame_get, sleepTimer)
time.sleep(max(peak_frame_time - duration, 0))
sleepTimer -= 0.0001
peak_time_between = max(peak_time_between, time.perf_counter() -frame_start)
total_run_time += time.perf_counter() -frame_start
@ -182,6 +182,7 @@ try:
packetQueue.put(packet)
else:
print("No data sent to VTS")
print(time.perf_counter() - totalFrameLatency)
except KeyboardInterrupt:
if not silent:

View file

@ -16,5 +16,5 @@ python3 -m venv .venv
source .venv/bin/activate
#the openseeface command
python3 facetracker.py --fps 30 --ip 127.0.0.1 --port 11573 --model 3 -W 1920 -H 1080 #-W 1920 -H 1080 #-W 640 -H 480 #--preview 1
python3 facetracker.py --fps 30 --ip 127.0.0.1 --port 11573 --model 3 -W 640 -H 480 #-W 1920 -H 1080 #-W 640 -H 480 #--preview 1

View file

@ -20,14 +20,12 @@ class Webcam():
self.cap = cv2.VideoCapture(0, cv2.CAP_DSHOW)
else:
self.cap = cv2.VideoCapture(0)
#tbh, these settings will make things way slower and frametimes will suffer
#but they're the only way frames bigger than 480p will work
if height > 480:
self.cap.set(cv2.CAP_PROP_FOURCC, cv2.VideoWriter_fourcc(*"MJPG"))
else:
self.cap.set(cv2.CAP_PROP_FOURCC, cv2.VideoWriter_fourcc(*"YUYV"))
self.cap.set(38, 2)
self.webcamBuffer = 2
self.cap.set(38, self.webcamBuffer)
self.cap.set(3, width)
self.cap.set(4, height)
self.cap.set(cv2.CAP_PROP_FPS, fps)
@ -46,6 +44,7 @@ class Webcam():
def start(self):
while self.cap.isOpened():
frameStart= time.perf_counter()
totalFrameLatency = time.perf_counter()
frame = self.getFrame()
self.cameraLatency = time.perf_counter() - frameStart
if self.ret:
@ -53,7 +52,7 @@ class Webcam():
if self.mirror:
frame = cv2.flip(frame, 1)
#this line is where this process sits 90% of the time
self.frameQueue.put([frame, self.cameraLatency])
self.frameQueue.put([frame, self.cameraLatency, totalFrameLatency])
if self.faceQueue.qsize() > 0:
self.updateGamma()
sleepTime = self.targetFrameTime - (time.perf_counter() - frameStart)