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

最新推荐文章于 2024-02-01 10:29:34 发布

微信公众号[机器学习炼丹术]

最新推荐文章于 2024-02-01 10:29:34 发布

阅读量3k

点赞数 1

分类专栏： python小知识 Kaggle实战笔记 PyTorch 从零学习深度网络文章标签： python tensorflow 大数据机器学习

本文链接：https://blog.csdn.net/qq_34107425/article/details/104164665

版权

71 篇文章 75 订阅

订阅专栏

49 篇文章 13 订阅

订阅专栏

28 篇文章 18 订阅

订阅专栏

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

#导入库
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
在这里插入图片描述
这是训练完成的样子：

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

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

在这里插入图片描述

关注

专栏目录