update

video_test.py

@@ -193,9 +193,9 @@ def main():
     # 初始化模型实例
     # TODO:修改模型网络名称/模型权重路径/视频路径
     predictor = Predictor(model_name='shufflenet-x2',
-                          weights_path=r'./shufflenet.pth',
+                          weights_path=r'./shufflenet-x2.pth',
                           num_classes=2)
-    input_path = r'D:\code\water_turbidity_det\tem_test\4_ch26_20260113075612_1'
+    input_path = r'D:\code\water_turbidity_det\tem_test\2_ch52_20260113011503_0'
     # 预处理图像
     all_imgs = os.listdir(input_path)
     all_imgs = [os.path.join(input_path, p) for p in all_imgs if p.split('.')[-1] in ['jpg', 'png']]