This is my Master thesis project at ETH Zurich, which contains two parts:
- Buildings Segmentation (see
buildings/Train.py), - Roads Segmentation (see
road_polygon_new/Train.py).
Data should be placed correctly (not in the folder data).
See Config.py file for more details.
If you want the data please send me an request email.
Arguments:
--city: the city name.--net: backbone of the net.--load: resume training or not.--vis: whether visualize the prediction or not.
To train the model just simply python3 Train.py --XXX YYY is fine.
To run inference just python3 Evaluate.py --XXX YYY.