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
......@@ -356,6 +356,7 @@ def fasterrcnn_resnet50_fpn(
>>> model = torchvision.models.detection.fasterrcnn_resnet50_fpn(pretrained=True)
>>> # For training
>>> 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))
>>> images = list(image for image in images)
>>> 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