从零开始做pytorch猫狗分类器(一)

这篇博客是PyTorch初学者的猫狗分类器教程,主要介绍了加载图片、log_softmax和ReLU方法、神经网络搭建、优化器和损失函数的使用,以及训练流程。在第二篇教程中将提供完整的代码实现。
摘要由CSDN通过智能技术生成

从零开始做pytorch猫狗分类器 (1)

对于一个pytorch初学者来说,迈进大门的第一步也许就是做一个猫狗分类器了,
可是网上的猫狗分类教程实在是有点少,而且一般都是直接上代码,让人没有独立思考的空间,本系列教程分2篇,第一篇(这一篇)来讲猫狗分类器的主要流程
第二篇就是上代码了。

加载图片

在这第一篇内,并不会讲到如何加载图片集,在第二篇教程中,我们将用ImageFolder和DataLoader来加载并打包成批处理文件,每张照片要统一为256*256像素,那么总结一下

用ImageFloader拿到照片
将照片统一规格为256*256
用DataLoader将图片打包

log_softmax和ReLU方法

我们不从数学的角度理解log_softmax方法,我们要清楚的知道,这个方法是把X个数值转换为百分比,百分比会随着一轮一轮的学习变得精确
ReLU方法的数学公式是 RuLU(x) = Max(0,x),他将零以上的部分保留,零以下的部分直接清零,如:
x = [0,-1,3,4,19,-78,-0.01,8]
ReLU(x) = [0,0,3,4,19,0,0,8]

搭建神

  • 10
    点赞
  • 28
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值