Pytorch-resnet18 猫狗识别

Pytorch-resnet18 猫狗识别

  • Pytorch-resnet18 猫狗识别
    • 数据集准备
        • 数据集下载
        • 数据集分类
    • 代码编写
    • 理论回顾

Pytorch-resnet18 猫狗识别

由于课程需要,完成一次“基于深度学习的猫狗识别”答辩,同时记录自己的研究和写代码过程,还有对于基础知识的回顾。

数据集准备

数据集下载

  • 采用Kaggle比赛官方的数据集,数据集下载链接如下:公开数据集,超好用,一般公开的数据集都可以从上面下载。

数据集分类

  • 将下载的cat vs dog 压缩包解压

代码编写

理论回顾

  1. 全连接层:

    对全连接层的概念不够清晰,查找资料补全信息。

  2. 卷积层层数和通道的变化

    以resnet18网络为例,输入照片大小3*224*224 ,通过卷积层Conv1,输出图片大小为64*112*112

你可能感兴趣的:(Pytorch,深度学习,pytorch)