Unverified Commit d1e1aa12 authored by Jackson Liu's avatar Jackson Liu Committed by GitHub
Browse files

Update rpn.py (#2149)

parent fed843da
...@@ -458,7 +458,7 @@ class RegionProposalNetwork(torch.nn.Module): ...@@ -458,7 +458,7 @@ class RegionProposalNetwork(torch.nn.Module):
""" """
Arguments: Arguments:
images (ImageList): images for which we want to compute the predictions images (ImageList): images for which we want to compute the predictions
features (List[Tensor]): features computed from the images that are features (OrderedDict[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).
......
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