当前位置：首页 > news >正文

无人机航拍建筑物识别航拍建筑物分割数据集 AI模型

news 2026/6/24 10:22:08

无人机航拍数据集建筑物分割

项目基本信息

项目名称：Buildings_Recognition（建筑物识别）
项目性质：基于深度学习的建筑物分割系统，使用PyTorch框架和U-Net架构实现。
主要功能：从航拍或卫星图像中精确分割出建筑物区域。
项目来源：源于【AI入门系列】地球观察员：建筑物识别学习赛。

核心特点

使用EfficientNet-B4作为编码器的U-Net模型，兼顾精度和效率。
实现RLE（游程编码）格式的编解码，优化存储和传输效率。
包含完整的数据增强流水线，提升模型泛化能力。
采用Dice系数作为评估指标，更适合分割任务。
实现早停机制，防止过拟合并节省训练时间。

数据集相关

数据集内容（经过data.ipynb处理）：
- train/：训练集图像（JPG格式）。
- train.csv：训练集标签（包含图像文件名和RLE编码的建筑物mask）。
- test/：测试集图像。
- test.csv：测试集标签。
数据格式要求：
- 图像尺寸建议为512x512像素。
- 标签使用RLE编码存储，空字符串表示无建筑物。
- 文件路径使用制表符分隔。
数据集下载：提供原始数据（Buildings_Recognition-origin）和处理后数据（Buildings_Recognition-processed）的下载链接。

文件夹结构

Building ├── data.ipynb # 数据预处理和分析文件 ├── data\ # 数据文件夹 ├── README-data.md # 数据集说明文档 ├── data1\ # 原始数据文件夹 ├── images\ # README-data图像文件夹 ├── test.csv # 测试集CSV文件 ├── test\ # 测试集图像文件夹 ├── 006TE20QA8.jpg # 测试图像文件 ... ├── test_a\ # 测试集A文件夹 ├── train.csv # 训练集CSV文件 ├── train\ # 训练集图像文件夹 └── train_mask.csv # 原始训练集mask的RLE编码文件 ├── log\ # 训练日志文件夹 ├── output\ # 输出文件夹 ├── model\ # 模型文件 └── pic\ # 结果图片 ├── predict.py ├── requirements.txt ├── train.py └── README.md

功能实现逻辑

数据预处理：
- BuildingDataset类：自动处理文件名大小写问题、自动添加文件扩展名、处理缺失文件情况（创建空白图像）、将RLE解码转换为二值mask。
- 数据增强：随机水平/垂直翻转、随机旋转90度、随机亮度/对比度调整、标准化处理（使用ImageNet均值方差）。
模型架构：U-Net with EfficientNet-B4，编码器部分使用预训练的EfficientNet-B4，解码器部分实现标准的U-Net结构，输出单通道logits。
训练流程：
- 损失函数：Dice Loss（直接从logits计算）。
- 优化器：Adam（lr=0.001）。
- 早停机制：验证集Dice系数连续5个epoch不提升则停止。
- 模型保存：保存验证集表现最好的模型。
评估指标：Dice系数（衡量预测mask与真实mask的重叠程度）、可视化（展示原图、真实mask和预测mask的对比）。

快速开始

安装依赖：pip install -r requirements.txt。
训练模型：python train.py。
配置参数（在train.py中修改）：
- BATCH_SIZE：批次大小（默认8）。
- LR：学习率（默认0.001）。
- EPOCHS：最大训练轮数（默认50）。
- PATIENCE：早停耐心值（默认5）。