参考github地址:https://github.com/lonngxiang/character-recognition-fastai
创建一个文件放train数据,格式可以参考,数据后续越多越好
用的是jupter book写的:
%reload_ext autoreload
%autoreload 2
%matplotlib inlinefrom fastai.vision import *
from fastai.metrics import error_ratepath=Path(r"C:\Users\lonng\Desktop\fastai v3")path.ls()
pathpath_img = path/"ziti"
fnames = get_image_files(path_img)
fnames[:5]
pat = r'ziti/(.*?)-\d+.jpg$'data = ImageDataBunch.from_name_re(path_img, fnames, pat, ds_tfms=get_transforms(), size=14, bs=6, num_workers=0).normalize(imagenet_stats)data.show_batch(rows=3, figsize=(9,7))learn = cnn_learner(data, models.resnet34, metrics=error_rate)
defaults.device=torch.device('cuda')
learn.fit_one_cycle(5)learn.save("modle1")img_test=open_image(r"C:/Users/lonng/Desktop/fastai v3/ziti/heiti-2.jpg")
img_test
aa,bb,cc=learn.predict(img_test)
aa,bb,cclearn.lr_find()
learn.recorder.plot()