agent 2.2
Browse files
models/matanyone_loader.py
CHANGED
|
@@ -989,6 +989,15 @@ def process_stream(
|
|
| 989 |
log.error(error_msg, exc_info=True)
|
| 990 |
_emit_progress(progress_cb, -1, f"CLEANUP ERROR: {error_msg}")
|
| 991 |
raise MatAnyError(error_msg) from e
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 992 |
|
| 993 |
def _flush_chunk(self, frames_bgr, seed_1hw, alpha_writer, fg_writer):
|
| 994 |
"""
|
|
|
|
| 989 |
log.error(error_msg, exc_info=True)
|
| 990 |
_emit_progress(progress_cb, -1, f"CLEANUP ERROR: {error_msg}")
|
| 991 |
raise MatAnyError(error_msg) from e
|
| 992 |
+
finally:
|
| 993 |
+
# Ensure all resources are cleaned up
|
| 994 |
+
if 'cap' in locals() and cap is not None:
|
| 995 |
+
cap.release()
|
| 996 |
+
if 'alpha_writer' in locals() and alpha_writer is not None:
|
| 997 |
+
alpha_writer.release()
|
| 998 |
+
if 'fg_writer' in locals() and fg_writer is not None:
|
| 999 |
+
fg_writer.release()
|
| 1000 |
+
_safe_empty_cache()
|
| 1001 |
|
| 1002 |
def _flush_chunk(self, frames_bgr, seed_1hw, alpha_writer, fg_writer):
|
| 1003 |
"""
|