提交 75e12915 编写于 作者: G gaotingquan 提交者: Tingquan Gao

debug

上级 e31509ee
......@@ -83,4 +83,4 @@ class CommonDataset(Dataset):
@property
def class_num(self):
return len(set(self.images))
return len(set(self.labels))
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册