MT_GCNN

The source code of the manuscript entitle "Move and remove: Multi-task learning for building simplification in vector maps with a graph convolutional neural network"

Requirements

  • python 3.7
  • pytorch 1.11.0
  • torch-geometric 2.0.3

Usage

Config the input directory and hyperparameters in main.py and run it. In case of understanding the proposed MT_GCNN model, check it in models.py with the class BuildingGenModel