从零开始做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]