运行

将数据放在data下的SVS_ALL和geojson_ALL下，运行generete_csv.py，生成模型训练集、测试集。 运行train.py进行训练

模型权重

将权重下载后放在Checkpoints中 链接：https://pan.baidu.com/s/1BeBIt3_K8tdBnYBTTeRXMg?pwd=zrrs 提取码：zrrs

数据格式

- Checkpoints
    - new_weights.pth
    - vit4k_xs_dino.pth
- data
    - geojson_ALL
    - SVS_ALL
    - merged_file.csv
- models
- utils
...