remove encoded images from memory to prevent exhaustion

k4yt3x · Feb 13, 2022 · 6378a36 · 6378a36
1 parent 0d0fd70
commit 6378a36
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/video2x/encoder.py b/video2x/encoder.py
@@ -133,8 +133,12 @@ def run(self):
                     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