Unverified Commit c50d0fcc authored by Prabhat Roy's avatar Prabhat Roy Committed by GitHub
Browse files

Added paper references to detection models (#4383)

* Added paper references to detection models

* Ignore linter warning

* Break long line into two
parent cadb1681
......@@ -300,6 +300,9 @@ def fasterrcnn_resnet50_fpn(pretrained=False, progress=True,
"""
Constructs a Faster R-CNN model with a ResNet-50-FPN backbone.
Reference: `"Faster R-CNN: Towards Real-Time Object Detection with
Region Proposal Networks" <https://arxiv.org/abs/1506.01497>`_.
The input to the model is expected to be a list of tensors, each of shape ``[C, H, W]``, one for each
image, and should be in ``0-1`` range. Different images can have different sizes.
......
......@@ -278,6 +278,8 @@ def keypointrcnn_resnet50_fpn(pretrained=False, progress=True,
"""
Constructs a Keypoint R-CNN model with a ResNet-50-FPN backbone.
Reference: `"Mask R-CNN" <https://arxiv.org/abs/1703.06870>`_.
The input to the model is expected to be a list of tensors, each of shape ``[C, H, W]``, one for each
image, and should be in ``0-1`` range. Different images can have different sizes.
......
......@@ -271,6 +271,8 @@ def maskrcnn_resnet50_fpn(pretrained=False, progress=True,
"""
Constructs a Mask R-CNN model with a ResNet-50-FPN backbone.
Reference: `"Mask R-CNN" <https://arxiv.org/abs/1703.06870>`_.
The input to the model is expected to be a list of tensors, each of shape ``[C, H, W]``, one for each
image, and should be in ``0-1`` range. Different images can have different sizes.
......
......@@ -569,6 +569,8 @@ def retinanet_resnet50_fpn(pretrained=False, progress=True,
"""
Constructs a RetinaNet model with a ResNet-50-FPN backbone.
Reference: `"Focal Loss for Dense Object Detection" <https://arxiv.org/abs/1708.02002>`_.
The input to the model is expected to be a list of tensors, each of shape ``[C, H, W]``, one for each
image, and should be in ``0-1`` range. Different images can have different sizes.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment