添加了训练集

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(
@@ -53,10 +53,10 @@ def train():
 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()