未验证 提交 3308cd92 编写于 作者: S Soumith Chintala 提交者: GitHub

Merge pull request #41 from killthekitten/patch-2

Fix BoxList constructor call in README.md
......@@ -163,7 +163,7 @@ class MyDataset(object):
labels = torch.tensor([10, 20])
# create a BoxList from the boxes
boxlist = BoxList(boxes, size=image.size, mode="xyxy")
boxlist = BoxList(boxes, image.size, mode="xyxy")
# add the labels to the boxlist
boxlist.add_field("labels", labels)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册