run.sh

# demo
mkdir output
cd demo/
python demo.py \
	     --config-file ../configs/ade20k-150/maskformer_R50_bs16_160k.yaml \
	     --input ../datasets/ADEChallengeData2016/release_test/testing/ADE_test_00003481.jpg ../datasets/ADEChallengeData2016/release_test/testing/ADE_test_00003482.jpg \
	     --output ../output \
	     --opts MODEL.WEIGHTS ../pretrained_models/model_final_d8dbeb.pkl

# train
python train_net.py \
    --config-file configs/ade20k-150/maskformer_R50_bs16_160k.yaml \
    --num-gpus 2 SOLVER.IMS_PER_BATCH 16 SOLVER.BASE_LR 0.0001

# test
python train_net.py \
       --config-file configs/ade20k-150/maskformer_R50_bs16_160k.yaml \
       --eval-only OUTPUT_DIR output MODEL.WEIGHTS ./pretrained_models/model_final_d8dbeb.pkl

# visualize
python demo/visualize_data.py \
       --config-file configs/ade20k-150/Vis-ADE20K-150.yaml \
       --source annotation \
       --output output2