Skip to content
This repository has been archived by the owner on Oct 9, 2023. It is now read-only.

Commit

Permalink
Update docs/source/reference/object_detection.rst
Browse files Browse the repository at this point in the history
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
  • Loading branch information
kaushikb11 and Borda authored Feb 16, 2021
1 parent fd8cc9b commit 76fa0b0
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion docs/source/reference/object_detection.rst
Original file line number Diff line number Diff line change
Expand Up @@ -107,7 +107,12 @@ To tailor the object detector to your dataset, you would need to have it in `COC
Model
*****

By default, we use the `Faster R-CNN <https://arxiv.org/abs/1506.01497>`_ model with a ResNet-50 FPN backbone. We also support `RetinaNet <https://arxiv.org/abs/1708.02002>`_. The inputs could be images of different sizes. The model behaves differently for training and evaluation. For training, it expects both the input tensors as well as the targets. And during evaluation, it expects only the input tensors and returns predictions for each image. The predictions are a list of boxes, labels and scores.
By default, we use the `Faster R-CNN <https://arxiv.org/abs/1506.01497>`_ model with a ResNet-50 FPN backbone.
We also support `RetinaNet <https://arxiv.org/abs/1708.02002>`_.
The inputs could be images of different sizes.
The model behaves differently for training and evaluation.
For training, it expects both the input tensors as well as the targets. And during the evaluation, it expects only the input tensors and returns predictions for each image.
The predictions are a list of boxes, labels, and scores.

------

Expand Down

0 comments on commit 76fa0b0

Please sign in to comment.