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