Skip to content

GitLab

Commit 49859788 authored by lambdaflow's avatar lambdaflow Committed by Francisco Massa
Browse files

Fix rpn.py typo (#1276)

parent c6e41b5d
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
torchvision/models/detection/rpn.py
View file @ 49859788
...@@ -351,7 +351,7 @@ class RegionProposalNetwork(torch.nn.Module): ...@@ -351,7 +351,7 @@ class RegionProposalNetwork(torch.nn.Module):
Returns: Returns:
objectness_loss (Tensor) objectness_loss (Tensor)
box_loss (Tensor box_loss (Tensor)
""" """
sampled_pos_inds, sampled_neg_inds = self.fg_bg_sampler(labels) sampled_pos_inds, sampled_neg_inds = self.fg_bg_sampler(labels)
...@@ -384,7 +384,7 @@ class RegionProposalNetwork(torch.nn.Module): ...@@ -384,7 +384,7 @@ class RegionProposalNetwork(torch.nn.Module):
features (List[Tensor]): features computed from the images that are features (List[Tensor]): features computed from the images that are
used for computing the predictions. Each tensor in the list used for computing the predictions. Each tensor in the list
correspond to different feature levels correspond to different feature levels
targets (List[Dict[Tensor]): ground-truth boxes present in the image (optional). targets (List[Dict[Tensor]]): ground-truth boxes present in the image (optional).
If provided, each element in the dict should contain a field `boxes`, If provided, each element in the dict should contain a field `boxes`,
with the locations of the ground-truth boxes. with the locations of the ground-truth boxes.
......
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