基于DINOv3的语义分割仓库,主要测试对象为遥感数据集;分类头主要是基于UPerNet;使用数据是whu-building数据集。
主干backbone+upernet+focalloss+polywarmupadamw
torch环境参考DINOv3的环境配置Installation
- 拉取dinov3:https://github.com/facebookresearch/dinov3.git
- 下载DINOv3预训练权重:
链接:https://pan.quark.cn/s/e3e1c26c88b1
提取码:gnKs
- 下载whu-building数据集:https://gpcv.whu.edu.cn/data/building_dataset.html
- 配置train_cfg.yaml文件
python train_seg.py --cfg train_cfg.yaml