.. DO NOT EDIT. .. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY. .. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE: .. "build/examples_detection/demo_faster_rcnn.py" .. LINE NUMBERS ARE GIVEN BELOW. .. only:: html .. note:: :class: sphx-glr-download-link-note Click :ref:`here ` to download the full example code .. rst-class:: sphx-glr-example-title .. _sphx_glr_build_examples_detection_demo_faster_rcnn.py: 02. Predict with pre-trained Faster RCNN models ============================================== This article shows how to play with pre-trained Faster RCNN model. First let's import some necessary libraries: .. GENERATED FROM PYTHON SOURCE LINES 8-13 .. code-block:: default from matplotlib import pyplot as plt import gluoncv from gluoncv import model_zoo, data, utils .. GENERATED FROM PYTHON SOURCE LINES 14-25 Load a pretrained model ------------------------- Let's get an Faster RCNN model trained on Pascal VOC dataset with ResNet-50 backbone. By specifying ``pretrained=True``, it will automatically download the model from the model zoo if necessary. For more pretrained models, please refer to :doc:`../../model_zoo/index`. The returned model is a HybridBlock :py:class:`gluoncv.model_zoo.FasterRCNN` with a default context of `cpu(0)`. .. GENERATED FROM PYTHON SOURCE LINES 25-28 .. code-block:: default net = model_zoo.get_model('faster_rcnn_resnet50_v1b_voc', pretrained=True) .. rst-class:: sphx-glr-script-out Out: .. code-block:: none Downloading /root/.mxnet/models/faster_rcnn_resnet50_v1b_voc-447328d8.zip from https://apache-mxnet.s3-accelerate.dualstack.amazonaws.com/gluon/models/faster_rcnn_resnet50_v1b_voc-447328d8.zip... 0%| | 0/121887 [00:00` .. container:: sphx-glr-download sphx-glr-download-jupyter :download:`Download Jupyter notebook: demo_faster_rcnn.ipynb ` .. only:: html .. rst-class:: sphx-glr-signature `Gallery generated by Sphinx-Gallery `_