Pytorch-resnet18 猫狗识别
由于课程需要,完成一次“基于深度学习的猫狗识别”答辩,同时记录自己的研究和写代码过程,还有对于基础知识的回顾。
数据集准备
数据集下载
- 采用Kaggle比赛官方的数据集,数据集下载链接如下:公开数据集,超好用,一般公开的数据集都可以从上面下载。
数据集分类
- 将下载的cat vs dog 压缩包解压
代码编写
理论回顾
- 全连接层:
对全连接层的概念不够清晰,查找资料补全信息。
- 卷积层层数和通道的变化
以resnet18网络为例,输入照片大小3*224*224 ,通过卷积层Conv1,输出图片大小为64*112*112,