Skip to content

Commit 2dcac82

Browse files
authored
Fix incorrect model name that breaks semantic segmentation demo (#284)
1 parent c9d4a46 commit 2dcac82

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

experiments/segmentation/demo.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
import encoding
33

44
# Get the model
5-
model = encoding.models.get_model('fcn_resnet50_ade', pretrained=True).cuda()
5+
model = encoding.models.get_model('fcn_resnet50s_ade', pretrained=True).cuda()
66
model.eval()
77

88
# Prepare the image

0 commit comments

Comments
 (0)