/offical-IndexNet

This is the official implementation of the IndexNet.

Primary LanguagePythonApache License 2.0Apache-2.0

IndexNet

Fig8

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)