Unverified Commit 202ecfd5 authored by WuZhe's avatar WuZhe Committed by GitHub
Browse files

update paper link for FCOS refence (#6035)



* update paper link for FCOS refence

* remove 'updated version'
Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
parent b430ba68
...@@ -201,7 +201,10 @@ class FCOSClassificationHead(nn.Module): ...@@ -201,7 +201,10 @@ class FCOSClassificationHead(nn.Module):
class FCOSRegressionHead(nn.Module): class FCOSRegressionHead(nn.Module):
""" """
A regression head for use in FCOS. A regression head for use in FCOS, which combines regression branch and center-ness branch.
This can obtain better performance.
Reference: `FCOS: A simple and strong anchor-free object detector <https://arxiv.org/abs/2006.09214>`_.
Args: Args:
in_channels (int): number of channels of the input feature in_channels (int): number of channels of the input feature
...@@ -680,6 +683,7 @@ def fcos_resnet50_fpn( ...@@ -680,6 +683,7 @@ def fcos_resnet50_fpn(
Constructs a FCOS model with a ResNet-50-FPN backbone. Constructs a FCOS model with a ResNet-50-FPN backbone.
Reference: `FCOS: Fully Convolutional One-Stage Object Detection <https://arxiv.org/abs/1904.01355>`_. Reference: `FCOS: Fully Convolutional One-Stage Object Detection <https://arxiv.org/abs/1904.01355>`_.
`FCOS: A simple and strong anchor-free object detector <https://arxiv.org/abs/2006.09214>`_.
The input to the model is expected to be a list of tensors, each of shape ``[C, H, W]``, one for each 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. 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