【遥感影像分类】使用PaddleAPI搭建ResNet50实现遥感影像分类任务
本文围绕遥感分类任务展开，使用西北工业大学2016年发布的含45类土地利用类型的遥感影像数据集，构建RESISC45Dataset自定义数据集，搭建ResNet50模型，经训练、验证，模型精度达0.83左右，最后进行了模型预测与效果展示。
前言
①. 关于任务
免费影视、动漫、音乐、游戏、小说资源长期稳定更新！ 👉 点此立即查看 👈
遥感分类，是指根据不同的分类标志以及遥感探测及应用侧重的方面不同，将遥感分成不同的类型。遥感图像计算机分类的依据是遥感图像像素的相似度。常使用距离和相关系数来衡量相似度。常见的分类方法有：监督分类、非监督分类法。
按遥感平台的不同，可把遥感分为航天遥感、航空遥感和地面（近地）遥感。按探测的电磁波段不同，可分为可见光遥感，红外遥感，微波遥感等。
②. 关于数据集
该数据集是由西北工业大学于2016年发布，包含提取自Google Earth的45种土地利用类型的遥感影像
数据集包含45个类别文件夹，每个文件夹下对应各自700幅遥感影像，一共有31500幅。
影像文件为三通道、大小为256*256的jpg格式文件
数据准备
解压已预先划分好的数据集
In [2]# 解压数据集!unzip -oq /home/aistudio/data/data131697/NWPU-RESISC45.zip登录后复制 In [3]
# 查看数据集文件结构!tree NWPU-RESISC45 -L 1登录后复制
自定义数据集
In [1]# 导入包import paddlefrom PIL import Imageimport osimport numpy as npimport random# 打印paddle版本print(paddle.__version__)登录后复制
2.2.2登录后复制 In [3]
class RESISC45Dataset(paddle.io.Dataset): def __init__(self, mode='train', label_path='NWPU-RESISC45/train_list.txt'): """ 初始化函数 """ assert mode in ['train', 'eval', 'test'], 'mode is one of train, eval, test.' self.mode = mode.lower() self.label_path = label_path self.data = [] with open(label_path) as f: for line in f.readlines(): info = line.strip().split(' ') if len(info) > 0: image_root = label_path.split('/')[0] info[0]=os.path.join(image_root,info[0]) self.data.append([info[0].strip(), info[1].strip()]) def preprocess(self,image): """ 数据增强函数 """ # 训练模式下的数据增强 if self.mode == 'train': # 裁剪大小 image = image.resize((224, 224), Image.BICUBIC) # 随机水平翻转 if random.randint(0, 1) == 1: image = image.transpose(Image.FLIP_LEFT_RIGHT) else: pass # 随机垂直翻转 if random.randint(0, 1) == 1: image = image.transpose(Image.FLIP_TOP_BOTTOM) else: pass # 图像归一化 image = np.asarray(image) image = image.astype('float32') mean = [0.485, 0.456, 0.406] std = [0.229, 0.224, 0.225] max_value = [255, 255, 255] min_value = [0, 0, 0] mean = np.asarray(mean, dtype=np.float32)[np.newaxis, np.newaxis, :] std = np.asarray(std, dtype=np.float32)[np.newaxis, np.newaxis, :] range_value = np.asarray([1. / (max_value[i] - min_value[i]) for i in range(len(max_value))],dtype=np.float32) image = (image - np.asarray(min_value, dtype=np.float32)) * range_value image -= mean image /= std # 数据格式转换 return paddle.to_tensor(image.transpose((2,0,1))) # 验证和测试模型下的数据增强 else: # 裁剪大小 image = image.resize((224, 224), Image.BICUBIC) # 图像归一化 image = np.asarray(image) image = image.astype('float32') mean = [0.485, 0.456, 0.406] std = [0.229, 0.224, 0.225] max_value = [255, 255, 255] min_value = [0, 0, 0] mean = np.asarray(mean, dtype=np.float32)[np.newaxis, np.newaxis, :] std = np.asarray(std, dtype=np.float32)[np.newaxis, np.newaxis, :] range_value = np.asarray([1. / (max_value[i] - min_value[i]) for i in range(len(max_value))],dtype=np.float32) image = (image - np.asarray(min_value, dtype=np.float32)) * range_value image -= mean image /= std # 数据格式转换 return paddle.to_tensor(image.transpose((2,0,1))) def __getitem__(self, index): """ 根据索引获取单个样本 """ image_file, label = self.data[index] image = Image.open(image_file) # 图片通道对齐 if image.mode != 'RGB': image = image.convert('RGB') # 进行数据增强 image = self.preprocess(image) return image, np.array(label, dtype='int64') def __len__(self): """ 获取样本总数 """ return len(self.data)登录后复制 实例化数据集
In [4]train_dataset=RESISC45Dataset(mode='train', label_path='NWPU-RESISC45/train_list.txt')val_dataset=RESISC45Dataset(mode='eval',label_path='NWPU-RESISC45/val_list.txt')test_dataset=RESISC45Dataset(mode='test',label_path='NWPU-RESISC45/test_list.txt')登录后复制
模型搭建
这里搭建的模型是ResNet50，论文地址：Deep_Residual_Learning_for_Image_Recognition
ResNet介绍
ResNet（Residual Neural Network）由微软研究院的Kaiming He等四名华人提出，通过使用ResNet Unit成功训练出了152层的神经网络，并在ILSVRC2015比赛中取得冠军，在top5上的错误率为3.57%，同时参数量比VGGNet低，效果非常突出。ResNet的结构可以极快的加速神经网络的训练，模型的准确率也有比较大的提升。同时ResNet的推广性非常好，甚至可以直接用到InceptionNet网络中。
代码实现
In [13]import paddleimport paddle.nn as nnfrom paddle.nn import Conv2D, MaxPool2D, AdaptiveAvgPool2D, Linear, ReLU, BatchNorm2Dimport paddle.nn.functional as F# 定义卷积批归一化块class ConvBNLayer(paddle.nn.Layer): def __init__(self, in_channels, out_channels, kernel_size, stride=1, act=None): super(ConvBNLayer, self).__init__() # 创建卷积层 self._conv = Conv2D( in_channels=in_channels, out_channels=out_channels, kernel_size=kernel_size, stride=stride, padding=(kernel_size - 1) // 2, bias_attr=False) # 创建BatchNorm层 self._batch_norm = BatchNorm2D(out_channels) # 创建activate层 self.act = act def forward(self, inputs): y = self._conv(inputs) y = self._batch_norm(y) if self.act == 'relu': y = F.relu(x=y) return y # 定义残差块class Bottleneckblock(paddle.nn.Layer): def __init__(self, inplane, in_channel, out_channel, stride = 1, start = False): super(Bottleneckblock, self).__init__() self.stride = stride self.start = start self.conv0 = ConvBNLayer(in_channel, inplane, 1, stride = stride, act='relu') self.conv1 = ConvBNLayer(inplane, inplane, 3, stride=1, act='relu') self.conv2 = ConvBNLayer(inplane, out_channel, 1, stride=1, act=None) self.conv3 = ConvBNLayer(in_channel, out_channel, 1, stride = stride, act=None) self.relu = nn.ReLU() def forward(self, inputs): y = inputs x = self.conv0(inputs) x = self.conv1(x) x = self.conv2(x) if self.start: y = self.conv3(y) z = self.relu(x+y) return zclass Resnet50(paddle.nn.Layer): def __init__(self, num_classes=45): super().__init__() # stem layers self.stem = nn.Sequential( nn.Conv2D(3, out_channels=64, kernel_size=7, stride=2, padding=3), nn.BatchNorm2D(64), nn.ReLU(), nn.MaxPool2D(kernel_size=3, stride=2, padding=1)) # blocks self.layer1 = self.add_bottleneck_layer(3, 64, start = True) self.layer2 = self.add_bottleneck_layer(4, 128) self.layer3 = self.add_bottleneck_layer(6, 256) self.layer4 = self.add_bottleneck_layer(3, 512) # head layer self.avgpool = nn.AdaptiveAvgPool2D(1) self.classifier = nn.Linear(2048, num_classes) def add_bottleneck_layer(self, num, inplane, start = False): layer = [] if start: layer.append(Bottleneckblock(inplane, inplane, inplane*4, start = True)) else: layer.append(Bottleneckblock(inplane, inplane*2, inplane*4, stride = 2, start = True)) for i in range(num-1): layer.append(Bottleneckblock(inplane, inplane*4, inplane*4)) return nn.Sequential(*layer) def forward(self, inputs): x = self.stem(inputs) x = self.layer1(x) x = self.layer2(x) x = self.layer3(x) x = self.layer4(x) x = self.avgpool(x) x = x.flatten(1) x = self.classifier(x) return x登录后复制
实例化Resnet50并打印模型结构
In [14]resnet50 = Resnet50(num_classes=45)登录后复制
W0310 10:31:31.892053 141 device_context.cc:447] Please NOTE: device: 0, GPU Compute Capability: 7.0, Driver API Version: 11.0, Runtime API Version: 10.1W0310 10:31:31.896260 141 device_context.cc:465] device: 0, cuDNN Version: 7.6.登录后复制 In [15]
paddle.summary(resnet50, (1, 3, 224, 224))登录后复制
------------------------------------------------------------------------------- Layer (type) Input Shape Output Shape Param # =============================================================================== Conv2D-1 [[1, 3, 224, 224]] [1, 64, 112, 112] 9,472 BatchNorm2D-1 [[1, 64, 112, 112]] [1, 64, 112, 112] 256 ReLU-1 [[1, 64, 112, 112]] [1, 64, 112, 112] 0 MaxPool2D-1 [[1, 64, 112, 112]] [1, 64, 56, 56] 0 Conv2D-2 [[1, 64, 56, 56]] [1, 64, 56, 56] 4,096 BatchNorm2D-2 [[1, 64, 56, 56]] [1, 64, 56, 56] 256 ConvBNLayer-1 [[1, 64, 56, 56]] [1, 64, 56, 56] 0 Conv2D-3 [[1, 64, 56, 56]] [1, 64, 56, 56] 36,864 BatchNorm2D-3 [[1, 64, 56, 56]] [1, 64, 56, 56] 256 ConvBNLayer-2 [[1, 64, 56, 56]] [1, 64, 56, 56] 0 Conv2D-4 [[1, 64, 56, 56]] [1, 256, 56, 56] 16,384 BatchNorm2D-4 [[1, 256, 56, 56]] [1, 256, 56, 56] 1,024 ConvBNLayer-3 [[1, 64, 56, 56]] [1, 256, 56, 56] 0 Conv2D-5 [[1, 64, 56, 56]] [1, 256, 56, 56] 16,384 BatchNorm2D-5 [[1, 256, 56, 56]] [1, 256, 56, 56] 1,024 ConvBNLayer-4 [[1, 64, 56, 56]] [1, 256, 56, 56] 0 ReLU-2 [[1, 256, 56, 56]] [1, 256, 56, 56] 0 Bottleneckblock-1 [[1, 64, 56, 56]] [1, 256, 56, 56] 0 Conv2D-6 [[1, 256, 56, 56]] [1, 64, 56, 56] 16,384 BatchNorm2D-6 [[1, 64, 56, 56]] [1, 64, 56, 56] 256 ConvBNLayer-5 [[1, 256, 56, 56]] [1, 64, 56, 56] 0 Conv2D-7 [[1, 64, 56, 56]] [1, 64, 56, 56] 36,864 BatchNorm2D-7 [[1, 64, 56, 56]] [1, 64, 56, 56] 256 ConvBNLayer-6 [[1, 64, 56, 56]] [1, 64, 56, 56] 0 Conv2D-8 [[1, 64, 56, 56]] [1, 256, 56, 56] 16,384 BatchNorm2D-8 [[1, 256, 56, 56]] [1, 256, 56, 56] 1,024 ConvBNLayer-7 [[1, 64, 56, 56]] [1, 256, 56, 56] 0 ReLU-3 [[1, 256, 56, 56]] [1, 256, 56, 56] 0 Bottleneckblock-2 [[1, 256, 56, 56]] [1, 256, 56, 56] 0 Conv2D-10 [[1, 256, 56, 56]] [1, 64, 56, 56] 16,384 BatchNorm2D-10 [[1, 64, 56, 56]] [1, 64, 56, 56] 256 ConvBNLayer-9 [[1, 256, 56, 56]] [1, 64, 56, 56] 0 Conv2D-11 [[1, 64, 56, 56]] [1, 64, 56, 56] 36,864 BatchNorm2D-11 [[1, 64, 56, 56]] [1, 64, 56, 56] 256 ConvBNLayer-10 [[1, 64, 56, 56]] [1, 64, 56, 56] 0 Conv2D-12 [[1, 64, 56, 56]] [1, 256, 56, 56] 16,384 BatchNorm2D-12 [[1, 256, 56, 56]] [1, 256, 56, 56] 1,024 ConvBNLayer-11 [[1, 64, 56, 56]] [1, 256, 56, 56] 0 ReLU-4 [[1, 256, 56, 56]] [1, 256, 56, 56] 0 Bottleneckblock-3 [[1, 256, 56, 56]] [1, 256, 56, 56] 0 Conv2D-14 [[1, 256, 56, 56]] [1, 128, 28, 28] 32,768 BatchNorm2D-14 [[1, 128, 28, 28]] [1, 128, 28, 28] 512 ConvBNLayer-13 [[1, 256, 56, 56]] [1, 128, 28, 28] 0 Conv2D-15 [[1, 128, 28, 28]] [1, 128, 28, 28] 147,456 BatchNorm2D-15 [[1, 128, 28, 28]] [1, 128, 28, 28] 512 ConvBNLayer-14 [[1, 128, 28, 28]] [1, 128, 28, 28] 0 Conv2D-16 [[1, 128, 28, 28]] [1, 512, 28, 28] 65,536 BatchNorm2D-16 [[1, 512, 28, 28]] [1, 512, 28, 28] 2,048 ConvBNLayer-15 [[1, 128, 28, 28]] [1, 512, 28, 28] 0 Conv2D-17 [[1, 256, 56, 56]] [1, 512, 28, 28] 131,072 BatchNorm2D-17 [[1, 512, 28, 28]] [1, 512, 28, 28] 2,048 ConvBNLayer-16 [[1, 256, 56, 56]] [1, 512, 28, 28] 0 ReLU-5 [[1, 512, 28, 28]] [1, 512, 28, 28] 0 Bottleneckblock-4 [[1, 256, 56, 56]] [1, 512, 28, 28] 0 Conv2D-18 [[1, 512, 28, 28]] [1, 128, 28, 28] 65,536 BatchNorm2D-18 [[1, 128, 28, 28]] [1, 128, 28, 28] 512 ConvBNLayer-17 [[1, 512, 28, 28]] [1, 128, 28, 28] 0 Conv2D-19 [[1, 128, 28, 28]] [1, 128, 28, 28] 147,456 BatchNorm2D-19 [[1, 128, 28, 28]] [1, 128, 28, 28] 512 ConvBNLayer-18 [[1, 128, 28, 28]] [1, 128, 28, 28] 0 Conv2D-20 [[1, 128, 28, 28]] [1, 512, 28, 28] 65,536 BatchNorm2D-20 [[1, 512, 28, 28]] [1, 512, 28, 28] 2,048 ConvBNLayer-19 [[1, 128, 28, 28]] [1, 512, 28, 28] 0 ReLU-6 [[1, 512, 28, 28]] [1, 512, 28, 28] 0 Bottleneckblock-5 [[1, 512, 28, 28]] [1, 512, 28, 28] 0 Conv2D-22 [[1, 512, 28, 28]] [1, 128, 28, 28] 65,536 BatchNorm2D-22 [[1, 128, 28, 28]] [1, 128, 28, 28] 512 ConvBNLayer-21 [[1, 512, 28, 28]] [1, 128, 28, 28] 0 Conv2D-23 [[1, 128, 28, 28]] [1, 128, 28, 28] 147,456 BatchNorm2D-23 [[1, 128, 28, 28]] [1, 128, 28, 28] 512 ConvBNLayer-22 [[1, 128, 28, 28]] [1, 128, 28, 28] 0 Conv2D-24 [[1, 128, 28, 28]] [1, 512, 28, 28] 65,536 BatchNorm2D-24 [[1, 512, 28, 28]] [1, 512, 28, 28] 2,048 ConvBNLayer-23 [[1, 128, 28, 28]] [1, 512, 28, 28] 0 ReLU-7 [[1, 512, 28, 28]] [1, 512, 28, 28] 0 Bottleneckblock-6 [[1, 512, 28, 28]] [1, 512, 28, 28] 0 Conv2D-26 [[1, 512, 28, 28]] [1, 128, 28, 28] 65,536 BatchNorm2D-26 [[1, 128, 28, 28]] [1, 128, 28, 28] 512 ConvBNLayer-25 [[1, 512, 28, 28]] [1, 128, 28, 28] 0 Conv2D-27 [[1, 128, 28, 28]] [1, 128, 28, 28] 147,456 BatchNorm2D-27 [[1, 128, 28, 28]] [1, 128, 28, 28] 512 ConvBNLayer-26 [[1, 128, 28, 28]] [1, 128, 28, 28] 0 Conv2D-28 [[1, 128, 28, 28]] [1, 512, 28, 28] 65,536 BatchNorm2D-28 [[1, 512, 28, 28]] [1, 512, 28, 28] 2,048 ConvBNLayer-27 [[1, 128, 28, 28]] [1, 512, 28, 28] 0 ReLU-8 [[1, 512, 28, 28]] [1, 512, 28, 28] 0 Bottleneckblock-7 [[1, 512, 28, 28]] [1, 512, 28, 28] 0 Conv2D-30 [[1, 512, 28, 28]] [1, 256, 14, 14] 131,072 BatchNorm2D-30 [[1, 256, 14, 14]] [1, 256, 14, 14] 1,024 ConvBNLayer-29 [[1, 512, 28, 28]] [1, 256, 14, 14] 0 Conv2D-31 [[1, 256, 14, 14]] [1, 256, 14, 14] 589,824 BatchNorm2D-31 [[1, 256, 14, 14]] [1, 256, 14, 14] 1,024 ConvBNLayer-30 [[1, 256, 14, 14]] [1, 256, 14, 14] 0 Conv2D-32 [[1, 256, 14, 14]] [1, 1024, 14, 14] 262,144 BatchNorm2D-32 [[1, 1024, 14, 14]] [1, 1024, 14, 14] 4,096 ConvBNLayer-31 [[1, 256, 14, 14]] [1, 1024, 14, 14] 0 Conv2D-33 [[1, 512, 28, 28]] [1, 1024, 14, 14] 524,288 BatchNorm2D-33 [[1, 1024, 14, 14]] [1, 1024, 14, 14] 4,096 ConvBNLayer-32 [[1, 512, 28, 28]] [1, 1024, 14, 14] 0 ReLU-9 [[1, 1024, 14, 14]] [1, 1024, 14, 14] 0 Bottleneckblock-8 [[1, 512, 28, 28]] [1, 1024, 14, 14] 0 Conv2D-34 [[1, 1024, 14, 14]] [1, 256, 14, 14] 262,144 BatchNorm2D-34 [[1, 256, 14, 14]] [1, 256, 14, 14] 1,024 ConvBNLayer-33 [[1, 1024, 14, 14]] [1, 256, 14, 14] 0 Conv2D-35 [[1, 256, 14, 14]] [1, 256, 14, 14] 589,824 BatchNorm2D-35 [[1, 256, 14, 14]] [1, 256, 14, 14] 1,024 ConvBNLayer-34 [[1, 256, 14, 14]] [1, 256, 14, 14] 0 Conv2D-36 [[1, 256, 14, 14]] [1, 1024, 14, 14] 262,144 BatchNorm2D-36 [[1, 1024, 14, 14]] [1, 1024, 14, 14] 4,096 ConvBNLayer-35 [[1, 256, 14, 14]] [1, 1024, 14, 14] 0 ReLU-10 [[1, 1024, 14, 14]] [1, 1024, 14, 14] 0 Bottleneckblock-9 [[1, 1024, 14, 14]] [1, 1024, 14, 14] 0 Conv2D-38 [[1, 1024, 14, 14]] [1, 256, 14, 14] 262,144 BatchNorm2D-38 [[1, 256, 14, 14]] [1, 256, 14, 14] 1,024 ConvBNLayer-37 [[1, 1024, 14, 14]] [1, 256, 14, 14] 0 Conv2D-39 [[1, 256, 14, 14]] [1, 256, 14, 14] 589,824 BatchNorm2D-39 [[1, 256, 14, 14]] [1, 256, 14, 14] 1,024 ConvBNLayer-38 [[1, 256, 14, 14]] [1, 256, 14, 14] 0 Conv2D-40 [[1, 256, 14, 14]] [1, 1024, 14, 14] 262,144 BatchNorm2D-40 [[1, 1024, 14, 14]] [1, 1024, 14, 14] 4,096 ConvBNLayer-39 [[1, 256, 14, 14]] [1, 1024, 14, 14] 0 ReLU-11 [[1, 1024, 14, 14]] [1, 1024, 14, 14] 0 Bottleneckblock-10 [[1, 1024, 14, 14]] [1, 1024, 14, 14] 0 Conv2D-42 [[1, 1024, 14, 14]] [1, 256, 14, 14] 262,144 BatchNorm2D-42 [[1, 256, 14, 14]] [1, 256, 14, 14] 1,024 ConvBNLayer-41 [[1, 1024, 14, 14]] [1, 256, 14, 14] 0 Conv2D-43 [[1, 256, 14, 14]] [1, 256, 14, 14] 589,824 BatchNorm2D-43 [[1, 256, 14, 14]] [1, 256, 14, 14] 1,024 ConvBNLayer-42 [[1, 256, 14, 14]] [1, 256, 14, 14] 0 Conv2D-44 [[1, 256, 14, 14]] [1, 1024, 14, 14] 262,144 BatchNorm2D-44 [[1, 1024, 14, 14]] [1, 1024, 14, 14] 4,096 ConvBNLayer-43 [[1, 256, 14, 14]] [1, 1024, 14, 14] 0 ReLU-12 [[1, 1024, 14, 14]] [1, 1024, 14, 14] 0 Bottleneckblock-11 [[1, 1024, 14, 14]] [1, 1024, 14, 14] 0 Conv2D-46 [[1, 1024, 14, 14]] [1, 256, 14, 14] 262,144 BatchNorm2D-46 [[1, 256, 14, 14]] [1, 256, 14, 14] 1,024 ConvBNLayer-45 [[1, 1024, 14, 14]] [1, 256, 14, 14] 0 Conv2D-47 [[1, 256, 14, 14]] [1, 256, 14, 14] 589,824 BatchNorm2D-47 [[1, 256, 14, 14]] [1, 256, 14, 14] 1,024 ConvBNLayer-46 [[1, 256, 14, 14]] [1, 256, 14, 14] 0 Conv2D-48 [[1, 256, 14, 14]] [1, 1024, 14, 14] 262,144 BatchNorm2D-48 [[1, 1024, 14, 14]] [1, 1024, 14, 14] 4,096 ConvBNLayer-47 [[1, 256, 14, 14]] [1, 1024, 14, 14] 0 ReLU-13 [[1, 1024, 14, 14]] [1, 1024, 14, 14] 0 Bottleneckblock-12 [[1, 1024, 14, 14]] [1, 1024, 14, 14] 0 Conv2D-50 [[1, 1024, 14, 14]] [1, 256, 14, 14] 262,144 BatchNorm2D-50 [[1, 256, 14, 14]] [1, 256, 14, 14] 1,024 ConvBNLayer-49 [[1, 1024, 14, 14]] [1, 256, 14, 14] 0 Conv2D-51 [[1, 256, 14, 14]] [1, 256, 14, 14] 589,824 BatchNorm2D-51 [[1, 256, 14, 14]] [1, 256, 14, 14] 1,024 ConvBNLayer-50 [[1, 256, 14, 14]] [1, 256, 14, 14] 0 Conv2D-52 [[1, 256, 14, 14]] [1, 1024, 14, 14] 262,144 BatchNorm2D-52 [[1, 1024, 14, 14]] [1, 1024, 14, 14] 4,096 ConvBNLayer-51 [[1, 256, 14, 14]] [1, 1024, 14, 14] 0 ReLU-14 [[1, 1024, 14, 14]] [1, 1024, 14, 14] 0 Bottleneckblock-13 [[1, 1024, 14, 14]] [1, 1024, 14, 14] 0 Conv2D-54 [[1, 1024, 14, 14]] [1, 512, 7, 7] 524,288 BatchNorm2D-54 [[1, 512, 7, 7]] [1, 512, 7, 7] 2,048 ConvBNLayer-53 [[1, 1024, 14, 14]] [1, 512, 7, 7] 0 Conv2D-55 [[1, 512, 7, 7]] [1, 512, 7, 7] 2,359,296 BatchNorm2D-55 [[1, 512, 7, 7]] [1, 512, 7, 7] 2,048 ConvBNLayer-54 [[1, 512, 7, 7]] [1, 512, 7, 7] 0 Conv2D-56 [[1, 512, 7, 7]] [1, 2048, 7, 7] 1,048,576 BatchNorm2D-56 [[1, 2048, 7, 7]] [1, 2048, 7, 7] 8,192 ConvBNLayer-55 [[1, 512, 7, 7]] [1, 2048, 7, 7] 0 Conv2D-57 [[1, 1024, 14, 14]] [1, 2048, 7, 7] 2,097,152 BatchNorm2D-57 [[1, 2048, 7, 7]] [1, 2048, 7, 7] 8,192 ConvBNLayer-56 [[1, 1024, 14, 14]] [1, 2048, 7, 7] 0 ReLU-15 [[1, 2048, 7, 7]] [1, 2048, 7, 7] 0 Bottleneckblock-14 [[1, 1024, 14, 14]] [1, 2048, 7, 7] 0 Conv2D-58 [[1, 2048, 7, 7]] [1, 512, 7, 7] 1,048,576 BatchNorm2D-58 [[1, 512, 7, 7]] [1, 512, 7, 7] 2,048 ConvBNLayer-57 [[1, 2048, 7, 7]] [1, 512, 7, 7] 0 Conv2D-59 [[1, 512, 7, 7]] [1, 512, 7, 7] 2,359,296 BatchNorm2D-59 [[1, 512, 7, 7]] [1, 512, 7, 7] 2,048 ConvBNLayer-58 [[1, 512, 7, 7]] [1, 512, 7, 7] 0 Conv2D-60 [[1, 512, 7, 7]] [1, 2048, 7, 7] 1,048,576 BatchNorm2D-60 [[1, 2048, 7, 7]] [1, 2048, 7, 7] 8,192 ConvBNLayer-59 [[1, 512, 7, 7]] [1, 2048, 7, 7] 0 ReLU-16 [[1, 2048, 7, 7]] [1, 2048, 7, 7] 0 Bottleneckblock-15 [[1, 2048, 7, 7]] [1, 2048, 7, 7] 0 Conv2D-62 [[1, 2048, 7, 7]] [1, 512, 7, 7] 1,048,576 BatchNorm2D-62 [[1, 512, 7, 7]] [1, 512, 7, 7] 2,048 ConvBNLayer-61 [[1, 2048, 7, 7]] [1, 512, 7, 7] 0 Conv2D-63 [[1, 512, 7, 7]] [1, 512, 7, 7] 2,359,296 BatchNorm2D-63 [[1, 512, 7, 7]] [1, 512, 7, 7] 2,048 ConvBNLayer-62 [[1, 512, 7, 7]] [1, 512, 7, 7] 0 Conv2D-64 [[1, 512, 7, 7]] [1, 2048, 7, 7] 1,048,576 BatchNorm2D-64 [[1, 2048, 7, 7]] [1, 2048, 7, 7] 8,192 ConvBNLayer-63 [[1, 512, 7, 7]] [1, 2048, 7, 7] 0 ReLU-17 [[1, 2048, 7, 7]] [1, 2048, 7, 7] 0 Bottleneckblock-16 [[1, 2048, 7, 7]] [1, 2048, 7, 7] 0 AdaptiveAvgPool2D-1 [[1, 2048, 7, 7]] [1, 2048, 1, 1] 0 Linear-1 [[1, 2048]] [1, 45] 92,205 ===============================================================================Total params: 23,653,421Trainable params: 23,547,181Non-trainable params: 106,240-------------------------------------------------------------------------------Input size (MB): 0.57Forward/backward pass size (MB): 328.09Params size (MB): 90.23Estimated Total Size (MB): 418.89-------------------------------------------------------------------------------登录后复制
{'total_params': 23653421, 'trainable_params': 23547181}登录后复制 模型训练
训练准备
In [22]from paddle.optimizer import Momentumfrom paddle.optimizer.lr import CosineAnnealingDecayfrom paddle.regularizer import L2Decayfrom paddle.nn import CrossEntropyLossfrom paddle.metric import Accuracyimport math# 总训练轮数Epochs = 30# 数据集读取的批次大小Batch_size = 64# 每轮的训练步数Step_each_epoch = math.ceil(len(train_dataset.data)/Batch_size)# 配置学习率Lr=CosineAnnealingDecay(learning_rate=0.06, T_max=Step_each_epoch * Epochs)# 配置优化器Optimizer = Momentum(learning_rate=Lr, momentum=0.9, weight_decay=L2Decay(1e-4), parameters=resnet50.parameters())# 设置损失函数Loss_fn = CrossEntropyLoss()# 构建数据读取器 Train_loader = paddle.io.DataLoader(train_dataset, batch_size=Batch_size, shuffle=True)Val_loader = paddle.io.DataLoader(val_dataset, batch_size=Batch_size)登录后复制
正式训练
In [11]def train(model, epochs, train_loader, val_loader, optimizer, loss_fn): ''' 训练函数 ''' acc_history = [0] for epoch in range(epochs): model.train() # 训练模式 for batch_id, data in enumerate(train_loader()): # 读取批次数据 x_data = data[0] # 训练数据 y_data = data[1] # 训练数据标签 y_data = paddle.reshape(y_data, (-1, 1)) predicts = model(x_data) # 预测结果 loss = loss_fn(predicts, y_data) # 计算损失 loss.backward() # 反向传播 optimizer.step() # 更新参数 optimizer.clear_grad() # 梯度清零 print("[TRAIN] epoch: {}/{}, loss is: {}".format(epoch+1, epochs, loss.numpy())) model.eval() # 验证模式 loss_list = [] acc_list = [] for batch_id, data in enumerate(val_loader()): # 读取批次数据 x_data = data[0] # 验证数据 y_data = data[1] # 验证数据标签 y_data = paddle.reshape(y_data, (-1, 1)) predicts = model(x_data) # 预测结果 loss = loss_fn(predicts, y_data) # 计算损失 acc = paddle.metric.accuracy(predicts, y_data) # 计算精度 loss_list.append(np.mean(loss.numpy())) acc_list.append(np.mean(acc.numpy())) print("[EVAL] Finished, Epoch={}, loss={}, acc={}".format(epoch+1, np.mean(loss_list), np.mean(acc_list))) if acc_history[-1] < np.mean(acc_list): paddle.save(resnet50.state_dict(),'output/resnet50.pdparams'.format(epoch)) acc_history.append(np.mean(acc_list))登录后复制 In [16]# 进行训练train(resnet50, Epochs, Train_loader, Val_loader, Optimizer, Loss_fn)登录后复制
模型验证
In [12]通过下面的代码可以看出我们的模型达到了0.83左右的精度
def val(model, val_loader): ''' 验证函数 ''' model.eval() #验证模式 acc_list = [] for batch_id, data in enumerate(val_loader()): x_data = data[0] # 验证数据 y_data = data[1] # 验证数据标签 y_data = paddle.reshape(y_data, (-1, 1)) predicts = model(x_data) # 预测结果 acc = paddle.metric.accuracy(predicts, y_data) # 计算精度 acc_list.append(np.mean(acc.numpy())) print("Eval finished, acc={}".format(np.mean(acc_list)))登录后复制 In [13]# 加载保存的模型resnet50.set_state_dict(paddle.load('output/resnet50.pdparams'))# 进行验证val(resnet50,Val_loader)登录后复制 Eval finished, acc=0.8262536525726318登录后复制
模型预测
In [14]我们将模型预测的标签结果存入列表results下。
def test(model, test_loader): model.eval() result_list = [] for batch_id, data in enumerate(test_loader()): x_data = data[0] # 测试数据 predicts = model(x_data) # 测试数据标签 result_list.append(np.argmax(predicts.numpy(),axis=1)) # 存入列表 print("predict finished") return result_list登录后复制 In [15]# 加载测试数据集Test_loader = paddle.io.DataLoader(test_dataset, batch_size=64)# 加载保存的模型resnet50.set_state_dict(paddle.load('output/resnet50.pdparams'))# 进行预测results = test(resnet50,Test_loader)登录后复制 predict finished登录后复制
效果展示
In [18]# 导入包%matplotlib inlineimport numpy as npimport cv2import matplotlib.pyplot as plt# 搭建label.txt的映射列表test_list = []with open('NWPU-RESISC45/labels.txt', 'r') as labels: for line in labels: test_list.append(line.strip())# 画图fig, axs = plt.subplots(nrows=5, ncols=1,figsize=(20,20))for i in range(5): img = cv2.imread(test_dataset.data[i+10][0],1) # 读取图片 img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) # 通道转换 ax = axs[i] ax.get_yaxis().set_visible(False) ax.get_xaxis().set_visible(False) ax.imshow(img) #展示图片 ax.set_title('Real: %s \n Predict: %s'%(test_list[i+10],test_list[results[0][i+10]])) #展示原先标签和预测结果登录后复制 登录后复制
游乐网为非赢利性网站，所展示的游戏/软件/文章内容均来自于互联网或第三方用户上传分享，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系youleyoucom@outlook.com。
同类文章
我把 Anthropic 的 Harness 工程思想做成了一个 Skill
用AI写代码，难在哪儿？ 用AI生成代码本身并不难，真正的挑战在于让它稳定地交付一个真正可用的东西。这篇文章，我们就来聊聊Anthropic工程团队是如何破解这个难题的，以及我如何将这套方法论落地成了一个可以复用的实战工具。 用 AI 写代码有多难？不是写不出来难，是让它稳定交付可用的东西很难。这篇
时间：2026-04-06 16:53
沃尔玛、塔吉特等美国零售巨头拥抱 AI，明确用户需为购物助手出错担责
美国零售巨头拥抱AI新玩法：功能归我，风险归你？ 最近有件事挺有意思，美国那边的大型零售商们，正铆足了劲把AI往购物流程里塞。但你猜怎么着？一旦AI捅了娄子，买单的却很可能变成了消费者自己。 这不，就在当地时间4月5号，外媒Futurism的一篇报道就点破了这个现象。企业们一边热火朝天地推广AI功能
时间：2026-04-06 13:52
小米物流大件“当日达”服务上线 50 城
小米物流大家电“当日达”实现全国50城覆盖，上午11点前下单最快当日送达 对于大家电配送时效长的普遍困扰，小米物流带来了全新的解决方案。最新消息显示，小米旗下大件商品的“当日达”服务范围已成功拓展至全国50座重点城市。除了北京、上海、广州、深圳、杭州、成都等一线与新一线核心城市外，此次升级还囊括了天
时间：2026-04-06 11:57
为什么现在很多人觉得 OpenClaw 不好用
当前开源版本的定位 你得明白，当前的开源版本，本质上更偏向于一个**开发者工具链**，而非一个即开即用的完整产品。它的核心组件非常明确： 一个基于 Node js 的运行环境 (runtime) 一个网关 (gateway) 插件与技能 (plugins skills) JSON 配置文件 命令
时间：2026-04-06 11:02
WorkBuddy工具
好的，我已准备好作为您专属的 SEO 内容优化专家开始工作。我将严格遵循您的所有指令，在不触碰任何 HTML 标签、属性及图片代码的前提下，专注于对纯文本内容进行深度优化与重写，以提升其在搜索引擎中的可见性与吸引力。 我的核心工作流程是：首先，我会精准解析您提供的原始文章，确保核心事实与信息结构毫发
时间：2026-04-06 08:34
- 日榜
- 周榜
- 月榜
相关攻略
2015-03-10 11:25
2015-03-10 11:05
2021-08-04 13:30
2015-03-10 11:22
2015-03-10 12:39
2022-05-16 18:57
2025-05-23 13:43
2025-05-23 14:01
热门教程
- 游戏攻略
- 安卓教程
- 苹果教程
- 电脑教程
逆水寒 掌握碎梦角色的定位技巧
发布于 2026-04-06
《洛克王国世界》独角兽伊利斯叫什么-呼唤独角兽的名字怎么写的
发布于 2026-04-06
《洛克王国世界》找到向上的方法任务怎么做-风眠圣所找到向上的方法任务图文攻略
发布于 2026-04-06
《洛克王国世界》叶冕魔力猫怎么打-叶冕魔力猫打法技巧攻略
发布于 2026-04-06
《洛克王国世界》罗隐在哪里抓-罗隐捕捉位置图解
发布于 2026-04-06
大店小二元宝与银两优先使用攻略-资源合理分配技巧
发布于 2026-04-06
《洛克王国世界》给1只精灵使用1次技能石任务怎么做-怎样给精灵使用技能石
发布于 2026-04-06
《斗兽情报6》勇夺灰烬堡垒加冕为王！
发布于 2026-04-06
使命召唤黑色行动7粉碎者手炮性能如何-使命召唤黑色行动7粉碎者手炮怎么样
发布于 2026-04-06
1月27日和平精英更新了什么-和平精英1月27日有哪些重磅更新
发布于 2026-04-06
三角洲蛊不羁人生如何获取-三角洲蛊不羁人生怎样得到
发布于 2026-04-06
龙魂旅人爬塔如何搭配阵容-龙魂旅人爬塔阵容搭配方法
发布于 2026-04-06
元梦之星啾灵有啥作用-元梦之星啾灵的作用是什么
发布于 2026-04-06
方舟深渊模组全饲料怎么做-方舟深渊模组全饲料制作攻略
发布于 2026-04-06
燕云十六声牵丝玉PVP连招技巧-燕云十六声牵丝玉PVP如何连招
发布于 2026-04-06
卡牌迷境深渊模式玩法攻略-卡牌迷境深渊模式怎么玩
发布于 2026-04-06
win10如何解决字体显示乱码_win10字体显示乱码完整指南一文搞懂
发布于 2026-04-06
小米电视语音唤醒不了是没网吗？
发布于 2026-04-06
支持PCIe 5.0的主板有哪些适合AMD平台？
发布于 2026-04-06
节能饮水机调温度后多久生效
发布于 2026-04-06
苹果音响使用说明在哪找？
发布于 2026-04-06
亿田集成灶S8D怎么清洗油网？
发布于 2026-04-06
传真机怎么复印文件带日期？
发布于 2026-04-06
外星人笔记本开机键在左边还是右边？
发布于 2026-04-06
小米移动电源充电器二合一支持边充边放吗
发布于 2026-04-06
