### Python 图像识别猫狗分类教程 #### 使用PyTorch实现猫狗分类 通过使用深度学习框架PyTorch，可以构建卷积神经网络（CNN）来完成猫狗分类的任务。此方法利用Swanlab训练可视化工具以及Gradio库创建交互式的Demo界面[^1]。 以下是基于PyTorch的一个简单示例代码： ```python import torch from torchvision import datasets, transforms from torch.utils.data import DataLoader import torch.nn as nn import torch.optim as optim # 数据预处理 transform = transforms.Compose([ transforms.Resize((150, 150)), transforms.ToTensor(), ]) train_dataset = datasets.ImageFolder(root='path_to_train_data', transform=transform) test_dataset = datasets.ImageFolder(root='path_to_test_data', transform=transform) train_loader = DataLoader(dataset=train_dataset, batch_size=32, shuffle=True) test_loader = DataLoader(dataset=test_dataset, batch_size=32, shuffle=False) # 定义简单的CNN模型 class CNNModel(nn.Module): def __init__(self): super(CNNModel, self).__init__() self.conv_layer = nn.Sequential( nn.Conv2d(in_channels=3, out_channels=32, kernel_size=3), nn.ReLU(), nn.MaxPool2d(kernel_size=2), nn.Conv2d(in_channels=32, out_channels=64, kernel_size=3), nn.ReLU(), nn.MaxPool2d(kernel_size=2), ) self.fc_layer = nn.Sequential( nn.Linear(64 * 35 * 35, 512), nn.ReLU(), nn.Linear(512, 2) # 输出层有两个类别：猫和狗 ) def forward(self, x): x = self.conv_layer(x) x = x.view(-1, 64 * 35 * 35) x = self.fc_layer(x) return x model = CNNModel() criterion = nn.CrossEntropyLoss() optimizer = optim.Adam(model.parameters(), lr=0.001) # 训练过程 for epoch in range(10): # 假设训练10轮次 model.train() running_loss = 0.0 for images, labels in train_loader: optimizer.zero_grad() outputs = model(images) loss = criterion(outputs, labels) loss.backward() optimizer.step() running_loss += loss.item() print(f'Epoch [{epoch+1}/10], Loss: {running_loss/len(train_loader)}') print('Finished Training') ``` #### 利用ImageDataGenerator进行数据增强 为了提高模型的泛化能力，可以通过`ImageDataGenerator`类来进行实时的数据增强操作。该生成器能够自动将图片转换为浮点数张量并调整大小至指定尺寸 (如150×150)，同时还能应用随机旋转、翻转等变换以增加样本多样性[^2]。 下面是一个具体的配置例子： ```python from tensorflow.keras.preprocessing.image import ImageDataGenerator datagen = ImageDataGenerator(rescale=1./255, rotation_range=40, width_shift_range=0.2, height_shift_range=0.2, shear_range=0.2, zoom_range=0.2, horizontal_flip=True, fill_mode='nearest') train_generator = datagen.flow_from_directory(directory='path_to_train_dir', target_size=(150, 150), class_mode='binary', batch_size=20) ```