去掉了main()中的train()

2023-12-07 19:04:57 +08:00
parent 77f0005b79
commit d13561c25f
1 changed files with 0 additions and 2 deletions
@@ -51,8 +51,6 @@ def train():
    model.export('G:\\Users\\15819\\Desktop\\model01.pkl') # M20.02.pkl

 def main():
-    train()
-
    image_path = 'G:\\Users\\15819\\Desktop\\TrainSet\\SmallCar\\京M88888.jpg'
    pred_class, confidence = predict_image(image_path)
    print(f"图片类别： {pred_class}, 置信度： {confidence}")