使用python构建卷积神经网络实现猫狗的分类问题

使用python构建卷积神经网络实现猫狗的分类问题

需求：使用卷积神经网络实现猫狗的二分类问题。
使用的语言：python
使用的工具：pytorch、pyCharm
数据集：链接：https://pan.baidu.com/s/1zfPFfmAjQ6ERJSQ4zEQHbw
提取码：vkk5

1、导入数据集

# 导入包
import os		# 用于当前文件
# 用于构建神经网络
import torch		
from torch import nn, optim	
from torch.nn import functional as F
from torch.utils import data
from torchvision import datasets
from torchvision import transforms

ImageSize = 224     # 定义图片大小
train_dir = './train/'    # 定义训练集存储路径
test_dir = './test/'    # 定义测试集存储路径
# 定义数据集预处理方法
data_transform = transforms.Compose([
    transforms.Resize((ImageSize, ImageSize)),    # 重新调整图片大小
    transforms.ToTensor(),       # 将图片转换为Tensor数组
    transforms.Normalize((0.1307,), (0.3081,))		# 归一化操作
])

# datasets.ImageFolder(dir, transform)：导入数据集
# dir：数据集存储路径
# transform：数据集预处理方法
datasets_train = dataset