
PyTorch-CNN完成了kaggle猫狗大战(Dogs vs. Cats)任务。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
今天,我们着重探讨如何利用PyTorch的卷积神经网络(CNN)来参与Kaggle的“猫狗大战”竞赛。不失为说,现在就直接切入主题。首先,图像数据集的来源是Kaggle平台。在网站上,通过搜索“Dogs vs. Cats”,可以找到大量的相关图像集,并从中选取一个进行下载。我所使用的具体数据集包含训练集和测试集:训练集包含4000张猫的图片以及4000张狗的图片;测试集则包含1000张猫的图片和1000张狗的图片。PyTorch的版本为:(torch 1.3.1+cpu)和(torchvision 0.4.2+cpu)。接下来,我们将详细介绍以下步骤:1. 我们将重新定义用于自定义数据集的方法;2. 我们将构建我们的PyTorch CNN模型结构;3. 利用我们已经定义的自定义数据集,加载整个数据集;4. 创建CNN模型的实例;5. 最后,我们将定义损失函数...
全部评论 (0)
还没有任何评论哟~


