diff --git a/torchvision/models/detection/fcos.py b/torchvision/models/detection/fcos.py index 684b52100db..3544ea3117e 100644 --- a/torchvision/models/detection/fcos.py +++ b/torchvision/models/detection/fcos.py @@ -201,7 +201,10 @@ def forward(self, x: List[Tensor]) -> Tensor: 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 `_. Args: in_channels (int): number of channels of the input feature @@ -680,6 +683,7 @@ def fcos_resnet50_fpn( Constructs a FCOS model with a ResNet-50-FPN backbone. Reference: `FCOS: Fully Convolutional One-Stage Object Detection `_. + `FCOS: A simple and strong anchor-free object detector `_. 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.