milleniums / high-resolution-remote-sensing-semantic-segmentation-pytorch

PyTorch实现高分遥感语义分割（地物分类）

High-Resolution-Remote-Sensing-Semantic-Segmentation-PyTorch

遥感也许还不能预测未来，但它一定会告诉你过去和现在，以及一步步见证着未来。

Update soon：

• 膨胀预测
• 后处理
• 半监督方法：伪标签
• 加入tensorboardX可视化输出

TODO:

• 总结训练步骤、技巧
• 分享预训练模型

教研室项目预测图展示

使用TensorBoard查看各项指标以及可视化（img, GT, Pre）

• 本地：

tensorboard --logdir=<your_log_dir>

• 服务器：

#首先使用ssh登陆服务器并转发端口:
ssh -L 6006:127.0.0.1:6006 远程服务器用户名@远程服务器Ip
(ssh -L 本地端口:本地IP:远程端口 远程服务器用户名@远程服务器Ip)

#在服务器上运行tensorboard:
tensorboard --logdir=<your_log_dir>

#在浏览器里输入:
远程服务器Ip:6006

• 各项指标样例

• 可视化测试集样例（img, GT, Pre）

数据分享

• 场景分类

  2019年遥感图像稀疏表征智能分析竞赛

  百度网盘 链接:https://pan.baidu.com/s/1NFvL8KV4pRwumBah05GG8A 密码:x5k0

• 变化检测

  2019年遥感图像稀疏表征智能分析竞赛

  百度网盘 链接:https://pan.baidu.com/s/1u4WKF_rRysvc0tEnrYq0Sg 密码:4udm

• 语义分割

  2019年县域农业大脑AI挑战赛

  百度网盘 链接:https://pan.baidu.com/s/1VXMLTCZM10quebX6Wf169w 密码:cmt7

  2019年遥感图像稀疏表征智能分析竞赛

  百度网盘 链接:https://pan.baidu.com/s/1LiMjwotcnQus0HFntEiZog 密码:3obz

  GID

  官方下载链接

  百度网盘 链接:https://pan.baidu.com/s/1LM6WX6zgihzIlY5OJWGEFA 密码:n8n0

  • 数据集信息

    

    

  • 样例展示

Reference

https://github.com/kazuto1011/deeplab-pytorch
https://github.com/fregu856/deeplabv3
https://github.com/zijundeng/pytorch-semantic-segmentation
https://github.com/jwyang/dcn.pytorch
https://github.com/lin-honghui/tianchi_CountyAgriculturalBrain_top1?spm=5176.12282029.0.0.278c1bb8nTiHfq

Contact

author: hk
qq：3025531252
email: [email protected]

high-resolution-remote-sensing-semantic-segmentation-pytorch's Issues

报错 self.train_dataset = RSDataset(class_name, root=args.train_data_root, mode='train', sync_transforms=sync_transform) TypeError: __init__() got multiple values for argument 'root'

请问这个情况是为什么？

求分享训练步骤

很多文件png、jpg、pth缺少，很难训练

module 'sync_transforms' has no attribute 'Mixup'

self.mixup_transform = sync_transforms.Mixup()请问这个问题，是需要将Mixup()直接去掉嘛

加载断点path

    resnet.load_state_dict((torch.load("/root/data/others/yaoganbisai/code_6_7/models/pretrained_model/resnet50-19c8e357.pth")))

if args.model == 'deeplabv3_version_1'中，设置了path之后，会报错。错误在上面。这个resnet里面的pth是怎么生成的。

error: no module name gid_classes

from class_names import gid_classes

In the dataset.py file, the line 4:

from class_names import gid_classes

But the file class_names.py just have the def fifteen_classes and def five_classes functions, where are the gid_classes?

RSDataset类初始化

self.class_names = gid_classes()

class_names.py中并没有gid_classes

请问这个怎么训练自己的训练集？

请问这个怎么训练自己的训练集？

Mixup

HI，there is no mixup!

Mixup()

self.mixup_transform = sync_transforms.Mixup()

sync_transforms.py中没有Mixup()这个函数啊

请问训练之前的GID数据集应该如何处理

Palette.json文件是什么

请问这个怎么输出预测图片？

请问怎么获得您在github上贴的那种预测图？有代码参考嘛，我在test.py文件中获得的预测图结果不对

请问这个怎么输出预测图片？

训练了一步后会出现gpu现存不足?

原始图片大小：24482448
切片大小：512512
batch-size:16
GPU为3090，显存：24G

数据集下载

parser.add_argument('--train-data-root', type=str, default='../data_5_6_21/train')
parser.add_argument('--val-data-root', type=str, default='../data_5_6_21/val')
这个训练集、验证集在哪里？需要另外下载吗？ @milleniums

RuntimeError: CUDA error

RuntimeError: CUDA error: device-side assert triggered
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.
代码中错误定位在trainer.training(epoch)和writer.add_scalar('train_loss', train_loss.avg, curr_iter)请问应该如何更改