remove encoded images from memory to prevent exhaustion

2024-12-28 23:19:11 +00:00 · 2022-02-13 03:47:27 +00:00 · 2022-02-13 03:47:27 +00:00 · 6378a36d91
commit 6378a36d91
parent 0d0fd70a24
1 changed files with 4 additions and 0 deletions
--- a/video2x/encoder.py
+++ b/video2x/encoder.py
@ -133,8 +133,12 @@ class VideoEncoder(threading.Thread):
                    time.sleep(0.1)
                    continue

+                # send the image to FFmpeg for encoding
                self.encoder.stdin.write(image.tobytes())

+                # remove the image from memory
+                self.processed_frames[frame_index] = None
+
                with self.processed.get_lock():
                    self.processed.value += 1