agent 2.3
Browse files- models/matanyone_loader.py +10 -5
models/matanyone_loader.py
CHANGED
|
@@ -946,9 +946,11 @@ def process_stream(
|
|
| 946 |
|
| 947 |
finally:
|
| 948 |
# Cleanup resources
|
|
|
|
| 949 |
try:
|
| 950 |
-
if 'cap' in locals() and cap
|
| 951 |
-
cap.
|
|
|
|
| 952 |
if 'alpha_writer' in locals() and alpha_writer is not None:
|
| 953 |
if hasattr(alpha_writer, 'isOpened') and alpha_writer.isOpened():
|
| 954 |
alpha_writer.release()
|
|
@@ -992,11 +994,14 @@ def process_stream(
|
|
| 992 |
finally:
|
| 993 |
# Ensure all resources are cleaned up
|
| 994 |
if 'cap' in locals() and cap is not None:
|
| 995 |
-
cap
|
|
|
|
| 996 |
if 'alpha_writer' in locals() and alpha_writer is not None:
|
| 997 |
-
alpha_writer
|
|
|
|
| 998 |
if 'fg_writer' in locals() and fg_writer is not None:
|
| 999 |
-
fg_writer
|
|
|
|
| 1000 |
_safe_empty_cache()
|
| 1001 |
|
| 1002 |
def _flush_chunk(self, frames_bgr, seed_1hw, alpha_writer, fg_writer):
|
|
|
|
| 946 |
|
| 947 |
finally:
|
| 948 |
# Cleanup resources
|
| 949 |
+
# Cleanup resources in a single finally block
|
| 950 |
try:
|
| 951 |
+
if 'cap' in locals() and cap is not None:
|
| 952 |
+
if hasattr(cap, 'isOpened') and cap.isOpened():
|
| 953 |
+
cap.release()
|
| 954 |
if 'alpha_writer' in locals() and alpha_writer is not None:
|
| 955 |
if hasattr(alpha_writer, 'isOpened') and alpha_writer.isOpened():
|
| 956 |
alpha_writer.release()
|
|
|
|
| 994 |
finally:
|
| 995 |
# Ensure all resources are cleaned up
|
| 996 |
if 'cap' in locals() and cap is not None:
|
| 997 |
+
if hasattr(cap, 'release'):
|
| 998 |
+
cap.release()
|
| 999 |
if 'alpha_writer' in locals() and alpha_writer is not None:
|
| 1000 |
+
if hasattr(alpha_writer, 'release'):
|
| 1001 |
+
alpha_writer.release()
|
| 1002 |
if 'fg_writer' in locals() and fg_writer is not None:
|
| 1003 |
+
if hasattr(fg_writer, 'release'):
|
| 1004 |
+
fg_writer.release()
|
| 1005 |
_safe_empty_cache()
|
| 1006 |
|
| 1007 |
def _flush_chunk(self, frames_bgr, seed_1hw, alpha_writer, fg_writer):
|