Skip to content

GitLab

Commit 3d561039 authored by Gu-ni-kim's avatar Gu-ni-kim Committed by Francisco Massa
Browse files

Update faster_rcnn.py (#979) 

Add 'import torch' in example
parent 47db9638
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
torchvision/models/detection/faster_rcnn.py
View file @ 3d561039
...@@ -99,6 +99,7 @@ class FasterRCNN(GeneralizedRCNN): ...@@ -99,6 +99,7 @@ class FasterRCNN(GeneralizedRCNN):
Example:: Example::
>>> import torch
>>> import torchvision >>> import torchvision
>>> from torchvision.models.detection import FasterRCNN >>> from torchvision.models.detection import FasterRCNN
>>> from torchvision.models.detection.rpn import AnchorGenerator >>> from torchvision.models.detection.rpn import AnchorGenerator
......
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