Fixed out-of-date dataset name (zhanghang1989#292)

mymuli · Jul 5, 2020 · 2d5b5ef · 2d5b5ef
1 parent 4f0a87b
commit 2d5b5ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/source/model_zoo/segmentation.rst b/docs/source/model_zoo/segmentation.rst
@@ -245,7 +245,7 @@ Quick Demo
     predict = torch.max(output, 1)[1].cpu().numpy() + 1
 
     # Get color pallete for visualization
-    mask = encoding.utils.get_mask_pallete(predict, 'pcontext')
+    mask = encoding.utils.get_mask_pallete(predict, 'pascal_voc')
     mask.save('output.png')