UnknownObject/PythonWorkGroup-0
1
0
Fork 0
generated from UnknownObject/PublicRepoDemo
Code Issues Pull Requests Packages Projects Releases Wiki Activity

最终版本

Browse Source
This commit is contained in:
UnknownObject
2023-12-06 19:53:41 +08:00
parent 2f97072a55
commit 257fb99811
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ver3.py
+2 -4
View File
@@ -15,8 +15,7 @@ from PIL import Image
def open_image(image_path): def open_image(image_path):
img = Image.open(image_path) img = Image.open(image_path)
img_cvt = img.resize((460,460)) img_cvt = img.resize((460,460))
return img return img_cvt
def predict_image(image_path): def predict_image(image_path):
# 加载模型 # 加载模型
@@ -36,7 +35,6 @@ def predict_image(image_path):
def train(): def train():
data_path = Path('G:\\Users\\15819\\Desktop\\Images2') data_path = Path('G:\\Users\\15819\\Desktop\\Images2')
export_path = Path('G:\\Users\\15819\\Desktop')
blocks = (ImageBlock, CategoryBlock) blocks = (ImageBlock, CategoryBlock)
batch_size = 32 batch_size = 32
dls = DataBlock( dls = DataBlock(
@@ -53,7 +51,7 @@ def train():
model.export('G:\\Users\\15819\\Desktop\\model01.pkl') model.export('G:\\Users\\15819\\Desktop\\model01.pkl')
def main(): def main():
#train() train()
image_path = 'G:\\Users\\15819\\Desktop\\Images2\\SmallCar\\京M88888.jpg' image_path = 'G:\\Users\\15819\\Desktop\\Images2\\SmallCar\\京M88888.jpg'
pred_class, confidence = predict_image(image_path) pred_class, confidence = predict_image(image_path)