添加了训练集

ver3.py

@@ -34,7 +34,7 @@ def predict_image(image_path):
     return pred_class, confidence
 
 def train():
-    data_path = Path('G:\\Users\\15819\\Desktop\\Images2')
+    data_path = Path('G:\\Users\\15819\\Desktop\\TrainSet')
     blocks = (ImageBlock, CategoryBlock)
     batch_size = 32
     dls = DataBlock(
@@ -48,15 +48,15 @@ def train():
 
     model = vision_learner(dls, resnet34, metrics=error_rate)
     model.fine_tune(5, freeze_epochs=3) #5 - 训练的轮次， 3 - 冻结的轮次
-    model.export('G:\\Users\\15819\\Desktop\\model01.pkl')
+    model.export('G:\\Users\\15819\\Desktop\\model01.pkl') # M20.02.pkl
 
 def main():
     train()
 
-    image_path = 'G:\\Users\\15819\\Desktop\\Images2\\SmallCar\\京M88888.jpg'
+    image_path = 'G:\\Users\\15819\\Desktop\\TrainSet\\SmallCar\\京M88888.jpg'
     pred_class, confidence = predict_image(image_path)
     print(f"图片类别： {pred_class}, 置信度： {confidence}")
 
 
 if __name__ == '__main__':
-    main()
+    train()