Michael-Xiu / Scale-Adaptive-Network

Semantic Image Segmentation by Scale-Adaptive Networks

Semantic Image Segmentation by Scale-Adaptive Networks

This code is a implementation of the experiments in the paper Semantic Image Segmentation by Scale-Adaptive Networks. The code is developed based on the Caffe framework.

License

SAN is released under the MIT License (refer to the LICENSE file for details).

Installing dependencies

• caffe (deeplabv2 version): deeplabv2 caffe installation instructions are available at https://bitbucket.org/aquariusjay/deeplab-public-ver2. Note, you need to compile caffe with python wrapper and support for python layers. Then add the caffe python path into tools/findcaffe.py.

Training the SAN model

• Run:

      $ python tools/train.py --solver YOUR_SOLVER --weight IMAGENET_PRETRAINED_MODEL --gpu GPU_ID

The corresponding solver files and input image lists are put in config and list floders.