【入门级实战】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

这是训练完成的样子：

公众号回复【下载】有精选的免费机器学习学习资料。 公众号每天会更新一个机器学习、深度学习的小知识，都是面试官会问的知识点哦~

• 【机器学习的基础数学(PDF)】
• 【竞赛中的大数据处理流程(PDF)】
• 【如何做大数据的基础特征工程(PDF)】
• 【自然语言处理NLP的应用实践大合集(PDF)】
• 【python入门级教材(400页PDF)】

公众号每天会更新一个机器学习、深度学习的小知识，都是面试官会问的知识点哦~