weixin_57266891的博客

columns=4rows=1# 窗口的宽度是固定的，而高度根据列数和行数来自动计算，以确保图像按照指定的布局显示。使用 gridspec.GridSpec 创建一个子图的网格布局。使用 plt.subplot 创建一个子图，并根据 gs 中的索引 idx 来选择子图的位置。plt.imshow 显示图像，其中 image_batch.at(idx) 表示从图像批次中获取第 idx 张图像并在子图中显示。

【代码】卫星图像应用 - 洪水检测 数据预处理。

with torch.no_grad(): # 禁止梯度计算，因为在准确率计算中不需要梯度信息else:else:训练深度学习模型的函数。参数：model: 要训练的深度学习模型criterion: 损失函数optimizer: 优化器acc: 准确率计算函数xtrain: 训练数据ytrain: 训练标签xval: 验证数据yval: 验证标签save_file_name: 保存训练后模型权重的文件名n_epochs: 训练的总轮数（epochs）

Pytochimport torchvision.transforms as transforms # 提供了各种用于预处理图像的转换函数from sklearn.metrics import accuracy_score # 计算模型的准确率self.conv1 = nn.Conv2d(3, 32, kernel_size = 3, stride = 1, padding = 1) # 输入通道数为3（RGB图像），输出通道数为32，使用3x3的卷积核，步幅为1，填充为1。