-
- 面向遥感城市场景图像语义分割场景,项目采用类似 UNet 的 Transformer 深度学习算法来实现,项目适用于卫星图像、航空图像、无人机图像等。
- 项目细节 ==> 具体参见项目
README.md
conda create -n airs python=3.8
conda activate airs
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
pip install -r requirements.txt
-
- (2) 下载预训练权重,具体见项目
README
- (3) 数据预处理 + 数据增强
- (4) 执行训练
python train_supervision.py -c config/uavid/unetformer.py
python vaihingen_test.py -c config/vaihingen/dcswin.py -o fig_results/vaihingen/dcswin --rgb -t 'd4'
-
- https://download.csdn.net/download/weixin_42405819/89100754