EasyOCR模型文件

去掉了main()中的train()
添加了训练集
2023-12-07 20:03:07 +08:00 · 2023-12-07 19:04:57 +08:00 · 2023-12-07 19:03:01 +08:00 · 2023-12-07 19:02:47 +08:00
84 changed files with 4 additions and 6 deletions
@@ -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,13 @@ 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\\TrainSet\\SmallCar\\京M88888.jpg'
    image_path = 'G:\\Users\\15819\\Desktop\\Images2\\SmallCar\\京M88888.jpg'
    pred_class, confidence = predict_image(image_path)
    print(f"图片类别： {pred_class}, 置信度： {confidence}")
 if __name__ == '__main__':
-    main()
+    train()
Author	SHA1	Message	Date
UnknownObject	5528cec244	EasyOCR模型文件	2023-12-07 20:03:07 +08:00
UnknownObject	d13561c25f	去掉了main()中的train()	2023-12-07 19:04:57 +08:00
UnknownObject	77f0005b79	添加了训练集	2023-12-07 19:03:01 +08:00
UnknownObject	8f964a02a5	最终版本	2023-12-07 19:02:47 +08:00