提交 31feebc7 编写于 作者: A Alexander Kalinovsky 提交者: Waleed

fix variable name

上级 6699b541
......@@ -133,7 +133,7 @@ class CocoDataset(utils.Dataset):
path=os.path.join(image_dir, coco.imgs[i]['file_name']),
width=coco.imgs[i]["width"],
height=coco.imgs[i]["height"],
annotations=coco.loadAnns(coco.getAnnIds(imgIds=[i],catIds=clase_ids, iscrowd=False)))
annotations=coco.loadAnns(coco.getAnnIds(imgIds=[i],catIds=class_ids, iscrowd=False)))
if return_coco:
return coco
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册