Skip to content

Commit

Permalink
Update plot_repurposing_annotations.py
Browse files Browse the repository at this point in the history
Changed `tranforms` to `transforms`.
  • Loading branch information
Bhavay-2001 committed Jun 28, 2024
1 parent bf01bab commit 915ddb1
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions gallery/others/plot_repurposing_annotations.py
Original file line number Diff line number Diff line change
Expand Up @@ -149,8 +149,8 @@ def show(imgs):
model = fasterrcnn_resnet50_fpn(weights=weights, progress=False)
print(img.size())

tranforms = weights.transforms()
img = tranforms(img)
transforms = weights.transforms()
img = transforms(img)
target = {}
target["boxes"] = boxes
target["labels"] = labels = torch.ones((masks.size(0),), dtype=torch.int64)
Expand Down

0 comments on commit 915ddb1

Please sign in to comment.