/SCAttNet

Semantic Segmentation Network with Spatial and Channel Attention Mechanism for High-Resolution Remote Sensing Images

Primary LanguagePython

SCAttNet

Semantic Segmentation Network with Spatial and Channel Attention Mechanism for High-Resolution Remote Sensing Images

Prerequisites

tensorflow 1.4.0 python 2.7

  1. Please put your train image, train_label into the path ./dataset/train_img, ./dataset/train_label
  2. Run python train.py to get the checkpoint file
  3. Run python predict.py to get the predicted results.