Unverified Commit 300019dd authored by Francisco Massa's avatar Francisco Massa Committed by GitHub
Browse files

Fix lint following #1695 (#1713)

* Fix lint following #1695

* V2

* V3
parent 06cbdb5b
...@@ -318,16 +318,19 @@ def fasterrcnn_resnet50_fpn(pretrained=False, progress=True, ...@@ -318,16 +318,19 @@ def fasterrcnn_resnet50_fpn(pretrained=False, progress=True,
Example:: Example::
>>> model = torchvision.models.detection.fasterrcnn_resnet50_fpn(pretrained=True) >>> model = torchvision.models.detection.fasterrcnn_resnet50_fpn(pretrained=True)
>>> images,boxes,labels = torch.rand(4,3,600,1200), torch.rand(4,11,4), torch.rand(4,11) # For Training >>> # For training
>>> images, boxes = torch.rand(4, 3, 600, 1200), torch.rand(4, 11, 4)
>>> labels = torch.randint(1, 91, (4, 11))
>>> images = list(image for image in images) >>> images = list(image for image in images)
>>> targets = [] >>> targets = []
>>> for i in range(len(images)): >>> for i in range(len(images)):
>>> d = {} >>> d = {}
>>> d['boxes'] = boxes[i] >>> d['boxes'] = boxes[i]
>>> d['labels'] = labels[i].type(torch.int64) >>> d['labels'] = labels[i]
>>> targets.append(d) >>> targets.append(d)
>>> output = model(images,targets) >>> output = model(images, targets)
>>> model.eval() # For inference >>> # For inference
>>> model.eval()
>>> x = [torch.rand(3, 300, 400), torch.rand(3, 500, 400)] >>> x = [torch.rand(3, 300, 400), torch.rand(3, 500, 400)]
>>> predictions = model(x) >>> predictions = model(x)
......
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