Commit 65eedbd8 authored by vfdev's avatar vfdev Committed by Francisco Massa

Update README.md (#9)

Fix a small typo
parent 3b6b9e7f
...@@ -134,7 +134,7 @@ class MyDataset(object): ...@@ -134,7 +134,7 @@ class MyDataset(object):
labels = torch.tensor([10, 20]) labels = torch.tensor([10, 20])
# create a BoxList from the boxes # create a BoxList from the boxes
boxlist = Boxlist(boxes, size=image.size, mode="xyxy") boxlist = BoxList(boxes, size=image.size, mode="xyxy")
# add the labels to the boxlist # add the labels to the boxlist
boxlist.add_field("labels", labels) boxlist.add_field("labels", labels)
......
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