"src/vscode:/vscode.git/clone" did not exist on "84b82a6cb7045de687684c23caa7683d43a7cd9d"
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