【入门级实战】fastAI分类猫狗实战代码

记录代码，方便之后快捷搭建框架，喜欢的可以收藏

#导入库
from fastai import *
from fastai.vision import *

#下载原数据
path = untar_data(URLs.DOGS)
URLs.DOGS,path

# 生成fastai中专门的数据类ImageDateBunch
data = ImageDataBunch.from_folder(path,ds_tfms=get_transforms(),size=224).normalize(imagenet_stats)

# 查看一个batch的数据
data.show_batch(rows=3,figsize=(6,6))

# 创建cnn模型
learner = cnn_learner(data,models.resnet18,metrics=[accuracy])

# 训练模型
learner.fit_one_cycle(1,1e-3)

这是训练过程中的图，fastai交互界面感觉挺好看的，上面是epoch，下面是batch

这是训练完成的样子：