### 使用PyTorch实现猫狗分类的卷积神经网络 为了构建一个用于猫狗分类的卷积神经网络(CNN)，可以采用如下结构： #### 数据集准备 数据预处理阶段非常重要。对于图像分类任务，通常会应用一些增强技术来扩充训练样本数量并提高模型泛化能力。 ```python import torchvision.transforms as transforms from torchvision.datasets import ImageFolder from torch.utils.data import DataLoader transform = transforms.Compose([ transforms.Resize((224, 224)), transforms.ToTensor(), ]) train_dataset = ImageFolder(root='path_to_train_data', transform=transform) val_dataset = ImageFolder(root='path_to_val_data', transform=transform) train_loader = DataLoader(train_dataset, batch_size=32, shuffle=True) val_loader = DataLoader(val_dataset, batch_size=32, shuffle=False) ``` #### 定义CNN架构 这里定义了一个简单的CNN模型，该模型由多个卷积层组成，后面接有全连接层来进行最终预测。 ```python import torch.nn as nn import torch.nn.functional as F class Net(nn.Module): def __init__(self): super(Net, self).__init__() self.conv1 = nn.Conv2d(in_channels=3, out_channels=16, kernel_size=5, stride=1, padding=2) self.pool = nn.MaxPool2d(kernel_size=2, stride=2, padding=0) self.conv2 = nn.Conv2d(in_channels=16, out_channels=32, kernel_size=5, stride=1, padding=2) self.fc1 = nn.Linear(32 * 56 * 56, 120) self.fc2 = nn.Linear(120, 84) self.fc3 = nn.Linear(84, 2) def forward(self, x): x = self.pool(F.relu(self.conv1(x))) x = self.pool(F.relu(self.conv2(x))) x = x.view(-1, 32 * 56 * 56) x = F.relu(self.fc1(x)) x = F.relu(self.fc2(x)) x = self.fc3(x) return x ``` #### 训练过程设置 配置损失函数、优化器以及学习率调整策略等参数，并编写完整的训练循环逻辑[^1]。 ```python device = 'cuda' if torch.cuda.is_available() else 'cpu' model = Net().to(device) criterion = nn.CrossEntropyLoss() optimizer = optim.Adam(model.parameters(), lr=0.001) for epoch in range(num_epochs): running_loss = 0.0 for i, data in enumerate(train_loader, 0): inputs, labels = data[0].to(device), data[1].to(device) optimizer.zero_grad() outputs = model(inputs) loss = criterion(outputs, labels) loss.backward() optimizer.step() running_loss += loss.item() if i % 200 == 199: print(f'[Epoch {epoch + 1}, Batch {i + 1}] Loss: {running_loss / 200:.3f}') running_loss = 0.0 ``` 上述代码展示了如何利用PyTorch框架搭建一个基础版本的猫狗二分类CNN模型及其训练流程。当然，在实际项目开发过程中还需要考虑更多细节问题，比如更复杂的网络设计、正则化方法的应用等等。