### 使用PyTorch实现简单猫狗分类模型 以下是基于PyTorch构建的一个简单猫狗图像分类模型的完整示例。该模型采用卷积神经网络（CNN），并结合常见的数据预处理方法来提升性能。 #### 数据准备 为了训练模型，通常需要一个包含猫和狗图片的数据集。这里假设已经下载了一个标准的猫狗分类数据集，并将其分为训练集和验证集[^1]。 ```python import torch from torchvision import datasets, transforms from torch.utils.data import DataLoader # 定义数据转换操作 transform = transforms.Compose([ transforms.Resize((64, 64)), # 调整大小 transforms.ToTensor(), # 转换为张量 transforms.Normalize([0.5], [0.5]) # 归一化 ]) # 加载数据集 train_dataset = datasets.ImageFolder(root='path_to_train_data', transform=transform) val_dataset = datasets.ImageFolder(root='path_to_val_data', transform=transform) # 创建DataLoader train_loader = DataLoader(train_dataset, batch_size=32, shuffle=True) val_loader = DataLoader(val_dataset, batch_size=32, shuffle=False) ``` #### 构建模型 下面是一个简单的卷积神经网络结构，适用于二分类任务： ```python import torch.nn as nn import torch.nn.functional as F class SimpleCNN(nn.Module): def __init__(self): super(SimpleCNN, self).__init__() # 卷积层 self.conv1 = nn.Conv2d(in_channels=3, out_channels=16, kernel_size=3, stride=1, padding=1) self.conv2 = nn.Conv2d(in_channels=16, out_channels=32, kernel_size=3, stride=1, padding=1) # 池化层 self.pool = nn.MaxPool2d(kernel_size=2, stride=2) # 全连接层 self.fc1 = nn.Linear(32 * 16 * 16, 128) # 输入维度取决于前面的池化结果 self.fc2 = nn.Linear(128, 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 * 16 * 16) # 展平 x = F.relu(self.fc1(x)) x = self.fc2(x) return x model = SimpleCNN() print(model) ``` #### 初始化权重 通过自定义函数`weights_init`初始化模型中的参数[^2]。 ```python def weights_init(m): if isinstance(m, nn.Conv2d): # 如果是卷积层 nn.init.xavier_uniform_(m.weight) if m.bias is not None: nn.init.zeros_(m.bias) elif isinstance(m, nn.Linear): # 如果是全连接层 nn.init.normal_(m.weight, mean=0, std=0.01) nn.init.zeros_(m.bias) model.apply(weights_init) ``` #### 训练过程 设置损失函数、优化器以及训练循环逻辑。 ```python criterion = nn.CrossEntropyLoss() # 多分类交叉熵损失 optimizer = torch.optim.Adam(model.parameters(), lr=0.001) device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') model.to(device) num_epochs = 10 for epoch in range(num_epochs): model.train() running_loss = 0.0 for images, labels in train_loader: images, labels = images.to(device), labels.to(device) optimizer.zero_grad() outputs = model(images) loss = criterion(outputs, labels) loss.backward() optimizer.step() running_loss += loss.item() print(f"Epoch [{epoch+1}/{num_epochs}], Loss: {running_loss/len(train_loader)}") print("Training complete.") ``` #### 验证模型 评估模型在验证集上的表现。 ```python correct = 0 total = 0 model.eval() with torch.no_grad(): for images, labels in val_loader: images, labels = images.to(device), labels.to(device) outputs = model(images) _, predicted = torch.max(outputs.data, 1) total += labels.size(0) correct += (predicted == labels).sum().item() accuracy = 100 * correct / total print(f"Validation Accuracy: {accuracy}%") ``` --- #### 总结 以上代码展示了如何利用PyTorch快速搭建一个用于猫狗分类的卷积神经网络模型。整个流程涵盖了数据加载、模型设计、权重初始化、训练与验证等多个方面[^1][^2]。