.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "build/examples_segmentation/demo_deeplab.py"
.. LINE NUMBERS ARE GIVEN BELOW.

.. only:: html

    .. note::
        :class: sphx-glr-download-link-note

        Click :ref:`here <sphx_glr_download_build_examples_segmentation_demo_deeplab.py>`
        to download the full example code

.. rst-class:: sphx-glr-example-title

.. _sphx_glr_build_examples_segmentation_demo_deeplab.py:

3. Test with DeepLabV3 Pre-trained Models
======================================

This is a quick demo of using GluonCV DeepLabV3 model on ADE20K dataset.
Please follow the `installation guide <../../index.html#installation>`__
to install MXNet and GluonCV if not yet.

.. GENERATED FROM PYTHON SOURCE LINES 8-16

.. code-block:: default

    import mxnet as mx
    from mxnet import image
    from mxnet.gluon.data.vision import transforms
    import gluoncv
    # using cpu
    ctx = mx.cpu(0)









.. GENERATED FROM PYTHON SOURCE LINES 17-21

Prepare the image
-----------------

download the example image

.. GENERATED FROM PYTHON SOURCE LINES 21-26

.. code-block:: default

    url = 'https://github.com/zhanghang1989/image-data/blob/master/encoding/' + \
        'segmentation/ade20k/ADE_val_00001755.jpg?raw=true'
    filename = 'ade20k_example.jpg'
    gluoncv.utils.download(url, filename, True)





.. rst-class:: sphx-glr-script-out

 Out:

 .. code-block:: none

    Downloading ade20k_example.jpg from https://github.com/zhanghang1989/image-data/blob/master/encoding/segmentation/ade20k/ADE_val_00001755.jpg?raw=true...

      0%|          | 0/13 [00:00<?, ?KB/s]
    14KB [00:00, 8474.56KB/s]             




.. GENERATED FROM PYTHON SOURCE LINES 27-28

load the image

.. GENERATED FROM PYTHON SOURCE LINES 28-34

.. code-block:: default

    img = image.imread(filename)

    from matplotlib import pyplot as plt
    plt.imshow(img.asnumpy())
    plt.show()




.. image-sg:: /build/examples_segmentation/images/sphx_glr_demo_deeplab_001.png
   :alt: demo deeplab
   :srcset: /build/examples_segmentation/images/sphx_glr_demo_deeplab_001.png
   :class: sphx-glr-single-img





.. GENERATED FROM PYTHON SOURCE LINES 35-36

normalize the image using dataset mean

.. GENERATED FROM PYTHON SOURCE LINES 36-39

.. code-block:: default

    from gluoncv.data.transforms.presets.segmentation import test_transform
    img = test_transform(img, ctx)








.. GENERATED FROM PYTHON SOURCE LINES 40-44

Load the pre-trained model and make prediction
----------------------------------------------

get pre-trained model

.. GENERATED FROM PYTHON SOURCE LINES 44-46

.. code-block:: default

    model = gluoncv.model_zoo.get_model('deeplab_resnet101_ade', pretrained=True)





.. rst-class:: sphx-glr-script-out

 Out:

 .. code-block:: none

    Downloading /root/.mxnet/models/deeplab_resnet101_ade-bf1584df.zip from https://apache-mxnet.s3-accelerate.dualstack.amazonaws.com/gluon/models/deeplab_resnet101_ade-bf1584df.zip...

      0%|          | 0/221969 [00:00<?, ?KB/s]
      0%|          | 101/221969 [00:00<04:36, 803.10KB/s]
      0%|          | 520/221969 [00:00<01:37, 2265.47KB/s]
      1%|          | 2182/221969 [00:00<00:30, 7234.27KB/s]
      3%|3         | 7550/221969 [00:00<00:09, 23029.09KB/s]
      7%|6         | 14610/221969 [00:00<00:05, 38583.15KB/s]
     10%|#         | 22401/221969 [00:00<00:03, 51093.04KB/s]
     14%|#3        | 30196/221969 [00:00<00:03, 59484.55KB/s]
     17%|#7        | 38051/221969 [00:00<00:02, 65368.22KB/s]
     21%|##        | 45798/221969 [00:00<00:02, 69068.17KB/s]
     24%|##4       | 53665/221969 [00:01<00:02, 71986.34KB/s]
     28%|##7       | 61377/221969 [00:01<00:02, 73348.15KB/s]
     31%|###1      | 69673/221969 [00:01<00:01, 76241.63KB/s]
     35%|###4      | 77347/221969 [00:01<00:01, 75721.66KB/s]
     39%|###8      | 85500/221969 [00:01<00:01, 77453.06KB/s]
     42%|####2     | 93272/221969 [00:01<00:01, 77342.55KB/s]
     46%|####5     | 101383/221969 [00:01<00:01, 78466.44KB/s]
     49%|####9     | 109244/221969 [00:01<00:01, 78259.35KB/s]
     53%|#####2    | 117167/221969 [00:01<00:01, 78548.28KB/s]
     56%|#####6    | 125174/221969 [00:01<00:01, 79002.29KB/s]
     60%|#####9    | 133080/221969 [00:02<00:01, 78589.53KB/s]
     64%|######3   | 141116/221969 [00:02<00:01, 78995.23KB/s]
     67%|######7   | 149019/221969 [00:02<00:00, 78701.69KB/s]
     71%|#######   | 156892/221969 [00:02<00:00, 78601.97KB/s]
     74%|#######4  | 164934/221969 [00:02<00:00, 79142.10KB/s]
     78%|#######7  | 172850/221969 [00:02<00:00, 79039.16KB/s]
     82%|########1 | 180953/221969 [00:02<00:00, 79632.97KB/s]
     85%|########5 | 188918/221969 [00:02<00:00, 62789.97KB/s]
     88%|########8 | 196298/221969 [00:02<00:00, 65563.78KB/s]
     92%|#########1| 203297/221969 [00:03<00:00, 66521.33KB/s]
     95%|#########4| 210326/221969 [00:03<00:00, 67551.98KB/s]
     98%|#########7| 217316/221969 [00:03<00:00, 67283.14KB/s]
    100%|##########| 221969/221969 [00:03<00:00, 65635.86KB/s]




.. GENERATED FROM PYTHON SOURCE LINES 47-48

make prediction using single scale

.. GENERATED FROM PYTHON SOURCE LINES 48-51

.. code-block:: default

    output = model.predict(img)
    predict = mx.nd.squeeze(mx.nd.argmax(output, 1)).asnumpy()








.. GENERATED FROM PYTHON SOURCE LINES 52-53

Add color pallete for visualization

.. GENERATED FROM PYTHON SOURCE LINES 53-58

.. code-block:: default

    from gluoncv.utils.viz import get_color_pallete
    import matplotlib.image as mpimg
    mask = get_color_pallete(predict, 'ade20k')
    mask.save('output.png')








.. GENERATED FROM PYTHON SOURCE LINES 59-60

show the predicted mask

.. GENERATED FROM PYTHON SOURCE LINES 60-64

.. code-block:: default

    mmask = mpimg.imread('output.png')
    plt.imshow(mmask)
    plt.show()




.. image-sg:: /build/examples_segmentation/images/sphx_glr_demo_deeplab_002.png
   :alt: demo deeplab
   :srcset: /build/examples_segmentation/images/sphx_glr_demo_deeplab_002.png
   :class: sphx-glr-single-img






.. rst-class:: sphx-glr-timing

   **Total running time of the script:** ( 0 minutes  7.738 seconds)


.. _sphx_glr_download_build_examples_segmentation_demo_deeplab.py:


.. only :: html

 .. container:: sphx-glr-footer
    :class: sphx-glr-footer-example



  .. container:: sphx-glr-download sphx-glr-download-python

     :download:`Download Python source code: demo_deeplab.py <demo_deeplab.py>`



  .. container:: sphx-glr-download sphx-glr-download-jupyter

     :download:`Download Jupyter notebook: demo_deeplab.ipynb <demo_deeplab.ipynb>`


.. only:: html

 .. rst-class:: sphx-glr-signature

    `Gallery generated by Sphinx-Gallery <https://sphinx-gallery.github.io>`_