Fix incorrect model name that breaks semantic segmentation demo (zhan…

…ghang1989#284)
liyouxing · Jun 28, 2020 · 2dcac82 · 2dcac82
1 parent c9d4a46
commit 2dcac82
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/experiments/segmentation/demo.py b/experiments/segmentation/demo.py
@@ -2,7 +2,7 @@
 import encoding
 
 # Get the model
-model = encoding.models.get_model('fcn_resnet50_ade', pretrained=True).cuda()
+model = encoding.models.get_model('fcn_resnet50s_ade', pretrained=True).cuda()
 model.eval()
 
 # Prepare the image