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):
"""
Arguments:
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
correspond to different feature levels
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