### 使用MindSpore或PyTorch构建卷积神经网络（CNN） #### MindSpore中的CNN实现 在MindSpore中，可以通过定义`Cell`类来创建自定义的CNN模型。以下是一个简单的基于MNIST数据集的手写数字识别CNN模型示例： ```python import numpy as np from mindspore import nn, Tensor, context from mindspore.dataset import MnistDataset from mindspore.train import Model from mindspore.nn import Accuracy context.set_context(mode=context.GRAPH_MODE, device_target="CPU") class CNNModel(nn.Cell): def __init__(self): super(CNNModel, self).__init__() self.conv1 = nn.Conv2d(1, 32, kernel_size=5, stride=1, pad_mode='same') self.relu = nn.ReLU() self.max_pool2d = nn.MaxPool2d(kernel_size=2, stride=2) self.flatten = nn.Flatten() self.fc1 = nn.Dense(32 * 7 * 7, 128) self.fc2 = nn.Dense(128, 10) def construct(self, x): x = self.conv1(x) x = self.relu(x) x = self.max_pool2d(x) x = self.flatten(x) x = self.fc1(x) x = self.relu(x) x = self.fc2(x) return x def create_dataset(data_path, batch_size=32, repeat_size=1): dataset = MnistDataset(data_path) resize_height, resize_width = 32, 32 rescale = 1.0 / 255.0 shift = 0.0 rescale_nml = 1 / 0.3081 shift_nml = -1 * 0.1307 / 0.3081 transform_img = [ lambda img: ((img.astype(np.float32) * rescale + shift).reshape(resize_height, resize_width)), lambda img: (img * rescale_nml + shift_nml), lambda img: img[np.newaxis, :, :] ] data_set = dataset.map(input_columns="label", operations=lambda label: np.eye(10)[label]) data_set = data_set.map(input_columns="image", operations=transform_img) data_set = data_set.batch(batch_size=batch_size, drop_remainder=True) data_set = data_set.repeat(count=repeat_size) return data_set network = CNNModel() net_loss = nn.SoftmaxCrossEntropyWithLogits(sparse=False, reduction="mean") net_opt = nn.Momentum(network.trainable_params(), learning_rate=0.01, momentum=0.9) model = Model(network, net_loss, net_opt, metrics={"Accuracy": Accuracy()}) ds_train = create_dataset("./data/MNIST_Data/train/") model.train(5, ds_train, callbacks=None, dataset_sink_mode=False) ``` 此代码展示了如何使用MindSpore训练一个基本的CNN模型[^3]。 --- #### PyTorch中的CNN实现 以下是使用PyTorch实现相同功能的一个简单例子： ```python import torch import torchvision.transforms as transforms from torch.utils.data import DataLoader from torchvision.datasets import MNIST from torch import nn, optim # 定义CNN模型 class SimpleCNN(nn.Module): def __init__(self): super(SimpleCNN, self).__init__() self.layer1 = nn.Sequential( nn.Conv2d(1, 32, kernel_size=5, padding=2), nn.BatchNorm2d(32), nn.ReLU(), nn.MaxPool2d(2)) self.layer2 = nn.Sequential( nn.Conv2d(32, 64, kernel_size=5, padding=2), nn.BatchNorm2d(64), nn.ReLU(), nn.MaxPool2d(2)) self.fc = nn.Linear(7*7*64, 10) def forward(self, x): out = self.layer1(x) out = self.layer2(out) out = out.view(out.size(0), -1) out = self.fc(out) return out device = 'cuda' if torch.cuda.is_available() else 'cpu' # 数据预处理和加载 transform = transforms.Compose([transforms.ToTensor()]) train_data = MNIST(root='./data', train=True, download=True, transform=transform) test_data = MNIST(root='./data', train=False, download=True, transform=transform) train_loader = DataLoader(dataset=train_data, batch_size=64, shuffle=True) test_loader = DataLoader(dataset=test_data, batch_size=64, shuffle=False) # 初始化模型、损失函数和优化器 model = SimpleCNN().to(device) criterion = nn.CrossEntropyLoss() optimizer = optim.Adam(model.parameters(), lr=0.001) # 训练过程 for epoch in range(5): # 进行五轮迭代 model.train() running_loss = 0.0 correct = 0 total = 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() _, predicted = torch.max(outputs.data, 1) total += labels.size(0) correct += (predicted == labels).sum().item() accuracy = 100 * correct / total print(f"Epoch [{epoch+1}/5], Loss: {loss.item():.4f}, Accuracy: {accuracy:.2f}%") ``` 这段代码展示了一个完整的PyTorch CNN模型训练流程[^4]。 --- #### 总结对比 两种框架都支持灵活的模型定义方式以及高效的自动求导机制。然而，在实际开发过程中需要注意以下几点差异： - **API风格**：PyTorch采用动态图模式，默认情况下更接近于Python原生语法；而MindSpore默认使用静态图模式，适合大规模分布式计算。 - **生态支持**：PyTorch拥有更为成熟的社区资源和支持文档，第三方库也更加丰富。 - **性能表现**：对于某些特定硬件平台（如昇腾处理器），MindSpore可能具备更好的适配性和加速效果。 如果目标环境涉及华为系列AI芯片，则优先考虑选用MindSpore作为主要开发工具[^5]。 ---