Skip to content

Shen001/dinov3_segmentation

Repository files navigation

dinov3_segmentation

介绍

基于DINOv3的语义分割仓库,主要测试对象为遥感数据集;分类头主要是基于UPerNet;使用数据是whu-building数据集。

软件架构

主干backbone+upernet+focalloss+polywarmupadamw

安装教程

torch环境参考DINOv3的环境配置Installation

  1. 拉取dinov3:https://github.com/facebookresearch/dinov3.git
  2. 下载DINOv3预训练权重:
链接:https://pan.quark.cn/s/e3e1c26c88b1
提取码:gnKs
  1. 下载whu-building数据集:https://gpcv.whu.edu.cn/data/building_dataset.html
  2. 配置train_cfg.yaml文件

运行

python train_seg.py --cfg train_cfg.yaml

致谢

  1. DINOv3
  2. segdino
  3. semantic-segmentation

About

DINOv3+UperNet for segmentation

Topics

Resources

License

Stars

Watchers

Forks

Contributors

Languages