MogensR commited on
Commit
e295279
·
1 Parent(s): 7ec2f7a
Files changed (1) hide show
  1. 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.isOpened():
951
- cap.release()
 
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.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):
 
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):