This is the official implementation of the IndexNet.
- Code for building model:
IndexNetModel
- Code for data dataset、augmentation:
dataset
- Implementation of Deeplabv3+ segmentation model:
deeplabv3plus
- Other utils:
utils
- PyTorch-Lightning Model:
trainer.py
- Config:
config.py
Many Thanks to: solo-learn、DetCon、Pytorch-BYOL
Pre-train Dataset | Architecture | Epoch | Link |
---|---|---|---|
Potsdam | ResNet-50 | 100 | BaiDu (Code: tick) |
Potsdam | ResNet-50 | 400 | BaiDu (Code: ebfk) |
Pre-train Dataset | Architecture | Epoch | Link |
---|---|---|---|
LoveDA (Urban) | ResNet-50 | 100 | BaiDu (Code:m3qf) |
LoveDA (Urban) | ResNet-50 | 400 | BaiDu (Code:4pmg) |
Pre-train Dataset | Architecture | Epoch | Link |
---|---|---|---|
SeCo-100K | ResNet-50 | 200 | BaiDu (Code:g997) |