wuhuikai / FastFCN

FastFCN: Rethinking Dilated Convolution in the Backbone for Semantic Segmentation.
http://wuhuikai.me/FastFCNProject
Other
838 stars 148 forks source link

how to use test_single_image.py #66

Closed TheRunningHorse closed 4 years ago

TheRunningHorse commented 4 years ago

I use test_single_image_image.py by python -m experiments.segmentation.test_single_image --input-path path_to_image --save-path save_path --model-zoo encnet_resnet50_ade and i fail 1

TheRunningHorse commented 4 years ago

I added jpu ='JPU' in line 30 of test_single_image.py but a new error was raised 799631043

wuhuikai commented 4 years ago

--model-zoo is not supported.

TheRunningHorse commented 4 years ago

so how do I test on single image?

wuhuikai commented 4 years ago
CUDA_VISIBLE_DEVICES=0,1,2,3 python -m experiments.segmentation.test_single_image --dataset [pcontext|ade20k] \
    --model [encnet|deeplab|psp] --jpu [JPU|JPU_X] \
    --backbone [resnet50|resnet101] [--ms] --resume {MODEL} --input-path {INPUT} --save-path {OUTPUT}
TheRunningHorse commented 4 years ago

Thanks a lot