-
Notifications
You must be signed in to change notification settings - Fork 9
Open
Description
如果去掉 cluster
类进行评估 IoU 和 F1 的话,仅仅使用 pspnet-r101 就能在 Potsdam数据上的超越现有ViT预训练的UpertNet 架构。结果如下:
复现步骤:
数据准备,参照mmseg官方示例,如下
openxlab dataset get --dataset-repo OpenDataLab/ISPRS_Potsdam --target-path /DATA
# 实验中需要使用 ‘2_Ortho_RGB.zip’ 和 ‘5_Labels_all_noBoundary.zip’ 。 仅保留这两个文件
mim run mmseg dataset_converters:potsdam /DATA/OpenDataLab___ISPRS_Potsdam/raw/Potsdam # 会在当前路径下新建一个目录存放切好的数据 data/potsdam/potsdam ,将生成 3456 张图像用于训练和 2016 张图像用于验证。
详细信息看mmseg官方文档,这里提供跳转链接。数据制作部分 模型和日志定位到Potsdam部分
下载权重和配置文件
mim download mmseg --config pspnet_r18-d8_4xb4-80k_potsdam-512x512 --dest /DATA #下载权重和配置文件
mim download mmseg --config pspnet_r101-d8_4xb4-80k_potsdam-512x512 --dest /DATA
测试
mim test mmseg /DATA/Postdam/pspnet_r101-d8_4xb4-80k_potsdam-512x512.py --checkpoint /DATA/Postdam/pspnet_r101-d8_4x4_512x512_80k_potsdam_20211220_125612-aed036c4.pth --cfg-options test_evaluator.iou_metrics=mIoU,mFscore test_dataloader.dataset.data_root=data/potsdam/potsdam # 指定上面切好数据的地址和配置文件权重下载地址
结果如下
pspnet_r18 | pspnet_r101 | |
---|---|---|
指标 | ![]() |
![]() |
疑问:请问下作者 Potsdam这个数据集为什么在 6类和5类的训练下存在如此大的差异,也许是标签使用5_Labels_all_noBoundary.zip的原因?
Metadata
Metadata
Assignees
Labels
No labels