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

2d5b5ef7 · Zj Li · GitHub · 4f0a87b0 · 2d5b5ef7
Unverified Commit 2d5b5ef7 authored Jul 05, 2020 by Zj Li Committed by GitHub Jul 04, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

docs/source/model_zoo/segmentation.rst docs/source/model_zoo/segmentation.rst +1 -1

No files found.
--- 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')