IndexNet
![Fig8](https://pumpkintypora.oss-cn-shanghai.aliyuncs.com/imgFig8.png)
This is the official implementation of the IndexNet.
Code
- 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
Reference
Many Thanks to: solo-learn、DetCon、Pytorch-BYOL
Pre-trained model
Potsdam
Pre-train Dataset |
Architecture |
Epoch |
Link |
Potsdam |
ResNet-50 |
100 |
BaiDu (Code: tick) |
Potsdam |
ResNet-50 |
400 |
BaiDu (Code: ebfk) |
LoveDA
Pre-train Dataset |
Architecture |
Epoch |
Link |
LoveDA (Urban) |
ResNet-50 |
100 |
BaiDu (Code:m3qf) |
LoveDA (Urban) |
ResNet-50 |
400 |
BaiDu (Code:4pmg) |
SeCo
Pre-train Dataset |
Architecture |
Epoch |
Link |
SeCo-100K |
ResNet-50 |
200 |
BaiDu (Code:g997) |