Skip to content

GitLab

"...pytorch/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "7c465d20fc32aa9e1a632034c92a398e57aee107"
Unverified Commit 93ec8bfd authored by Mriganka Nath's avatar Mriganka Nath Committed by GitHub
Browse files

updated the example for fasterrcnn_resnet50 (#5165) 


Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
parent 80b0e35c
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
torchvision/models/detection/faster_rcnn.py
View file @ 93ec8bfd
...@@ -356,6 +356,7 @@ def fasterrcnn_resnet50_fpn( ...@@ -356,6 +356,7 @@ def fasterrcnn_resnet50_fpn(
>>> model = torchvision.models.detection.fasterrcnn_resnet50_fpn(pretrained=True) >>> model = torchvision.models.detection.fasterrcnn_resnet50_fpn(pretrained=True)
>>> # For training >>> # For training
>>> images, boxes = torch.rand(4, 3, 600, 1200), torch.rand(4, 11, 4) >>> images, boxes = torch.rand(4, 3, 600, 1200), torch.rand(4, 11, 4)
>>> boxes[:, :, 2:4] = boxes[:, :, 0:2] + boxes[:, :, 2:4]
>>> labels = torch.randint(1, 91, (4, 11)) >>> labels = torch.randint(1, 91, (4, 11))
>>> images = list(image for image in images) >>> images = list(image for image in images)
>>> targets = [] >>> targets = []
......
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