# remote-sensing-semantic-segmentation

**Repository Path**: zhang-sec/remote-sensing-semantic-segmentation

## Basic Information

- **Project Name**: remote-sensing-semantic-segmentation
- **Description**: 遥感影像的语义分割实验：FCN，UNet，SegNet，Capsule
- **Primary Language**: Python
- **License**: MIT
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 4
- **Created**: 2022-08-17
- **Last Updated**: 2022-08-17

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# remote-sensing-semantic-segmentation

#### 介绍
遥感影像的语义分割实验-建筑物分割，方法包括：FCN，UNet，SegNet，Capsule

#### 环境安装

```
conda install numpy
conda install pillow
conda install matplotlib
conda install shapely
conda install gdal
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
```

#### 数据集操作说明

data目录结构：
```
data
    --|rawdata--|img
              --|mask
              --|mask2png
              --|TDT_data
              --|fitname.py,mask2png.py,shp2tif.py

    --|train---|label
            ---|src

    --|val-----|label
          -----|src

```
- TDT_data为tif原图+shp标注；
- 运行顺序为：shp2tif.py，mask2png.py，fitname.py；
- train与val需手动划分

#### 训练过程
![输入图片说明](https://images.gitee.com/uploads/images/2021/0512/185558_8bdb6cb8_5644878.png "log_visual.png")
注意：胶囊网络的实现暂未开源

#### 推理结果
![输入图片说明](https://images.gitee.com/uploads/images/2021/0513/105421_91df9d61_5644878.png "结果对比.png")