Giter VIP home page Giter VIP logo

jstars-fgn-cap's Introduction

Feature guide network with Context Aggregation Pyramid for Remote Sensing Image Segmentation, JSTARS, 2022.

Jiaojiao Li, Yuzhe Liu, Jiacha Liu, Rui song, Wei Liu, Kailiang Han and Qian Du.


Code for the paper: Feature guide network with Context Aggregation Pyramid for Remote Sensing Image Segmentation.

Fig. 1: Our proposed network architecture. The network is based on encoder-decoder structure. In the encoder stage, edge and body information in the feature maps are extracted respectively constrained by corresponding loss function. The backbone is followed by a CAP to aggregate multi-scale context features adaptively. In the decoder stage, an EGFTM is used to help with upsampling to restore resolution.

Training and Test Process

  1. The data sets used in this article are Vaihingen, Potsdam, TianZhi, and Cityscapes.
  2. The network model file is placed in the 'models/net_work/final' folder and contains the ablation experiment network for each part.
  3. Run "train.py" to retrain the FGN-CAP results.
  4. Run "eval.py" to obtain the segmentation results of the training model
  5. "train_cityscapes.py" and "test_cityscape.py" are training and testing code designed for the cityscape dataset.

We have successfully tested it on Ubuntu 18.04 with PyTorch 1.12.0.

References

If you find this code helpful, please kindly cite:

[1]J. Li et al., "Feature Guide Network With Context Aggregation Pyramid for Remote Sensing Image Segmentation," in IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing, vol. 15, pp. 9900-9912, 2022, doi: 10.1109/JSTARS.2022.3221860.

Citation Details

BibTeX entry:

@article{li2022feature,
  title={Feature Guide Network With Context Aggregation Pyramid for Remote Sensing Image Segmentation},
  author={Li, Jiaojiao and Liu, Yuzhe and Liu, Jiachao and Song, Rui and Liu, Wei and Han, Kailiang and Du, Qian},
  journal={IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing},
  volume={15},
  pages={9900--9912},
  year={2022},
  publisher={IEEE}
}

Licensing

Copyright (C) 2022 Yuzhe Liu

This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, version 3 of the License.

This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License along with this program.

jstars-fgn-cap's People

Contributors

lyz123-xidian avatar

Watchers

 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.