From 257fb99811e677198a4beb5bd20defc287a413b2 Mon Sep 17 00:00:00 2001
From: UnknownObject <me@unsmail.ltd>
Date: Wed, 6 Dec 2023 19:53:41 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9C=80=E7=BB=88=E7=89=88=E6=9C=AC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ver3.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/ver3.py b/ver3.py
index aa284c0..5ce18dd 100644
--- a/ver3.py
+++ b/ver3.py
@@ -15,8 +15,7 @@ from PIL import Image
 def open_image(image_path):
     img = Image.open(image_path)
     img_cvt = img.resize((460,460))
-    return img
-
+    return img_cvt
 
 def predict_image(image_path):
     # 加载模型
@@ -36,7 +35,6 @@ def predict_image(image_path):
 
 def train():
     data_path = Path('G:\\Users\\15819\\Desktop\\Images2')
-    export_path = Path('G:\\Users\\15819\\Desktop')
     blocks = (ImageBlock, CategoryBlock)
     batch_size = 32
     dls = DataBlock(
@@ -53,7 +51,7 @@ def train():
     model.export('G:\\Users\\15819\\Desktop\\model01.pkl')
 
 def main():
-    #train()
+    train()
 
     image_path = 'G:\\Users\\15819\\Desktop\\Images2\\SmallCar\\京M88888.jpg'
     pred_class, confidence = predict_image(image_path)