Pytorch–CNN实现kaggle猫狗大战(Dogs vs. Cats)

今天咱们来聊聊用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