基于卷积神经网络的猫狗识别

一、环境配置

1、安装Anaconda
下载安装Anaconda
2、配置Pyorch

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torchvision

二、数据集

百度网盘下载：数据集
提取码：dmp4

kaggle网站的数据集下载地址：
添加链接描述

三、猫狗分类

1、导入库

import torch.nn.functional as F
import torch.optim as optim
import torch
import torch.nn as nn
import torch.nn.parallel
 
import torch.optim
import torch.utils.data
import torch.utils.data.distributed
import torchvision.transforms as transforms
import torchvision.datasets as datasets

2、设置超参数

#每次的个数
BATCH_SIZE = 20
#迭代次数
EPOCHS = 10
#采用cpu还是gpu进行计算
DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

3、数据预处理

 
transform = transforms.Compose([
    transforms.Resize(100),
    transfo