|
|
@@ -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']]
|
jiyuhang