fixed a bug again

2026-01-31 16:03:29 +08:00
parent 2d15e697d0
commit 8c626fa1c1
1 changed files with 2 additions and 0 deletions
--- a/video_cleaner.py
+++ b/video_cleaner.py
@@ -6,6 +6,8 @@ import cv2
 import numpy as np
 from pathlib import Path
 from datetime import datetime, timedelta
+from concurrent.futures import ProcessPoolExecutor, as_completed
+import torch

 # Try to import YOLO
 try: