IndexNet
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) |