今天咱们来聊聊用Pytorch的CNN完成kaggle猫狗大战。
话不多说,进入正题。
首先,图片数据来源kaggle,在网站上搜索Dogs vs. Cats很多相关图片集,找一个下载下来。
我这里采用的数据集是:
Train:4000张cat + 4000张dog
Test:1000张cat + 1000张dog
Pytorch版本:(torch 1.3.1+cpu) + (torchvision 0.4.2+cpu)
步骤:
1. 重定义我们的Dataset
2. 定义我们的Pytorch CNN结构
3. 利用定义好的Dataset,载入我们的数据集
4. 创建CNN实例
5. 定义lo
1