Original code is taken from https://github.com/xinntao/BasicSR
-
Download (
decoder_epoch_20.pthandencoder_epoch_20.pth) for segmentation ade20k from here and put them intoweights/ade20k-resnet50dilated-ppm_deepsupfolder -
Download pretrained weight (
G.pth) for our model from drive and put it intoweightsfolder -
Run
CUDA_VISIBLE_DEVICES=0 python3 src/predict.py -
All results are stored in
outputs