未验证 提交 684fe1b6 编写于 作者: W wangguanzhong 提交者: GitHub

support use_default_label for infer (#151)

上级 993b955b
......@@ -120,12 +120,11 @@ class ImageFolder(DataSet):
anno_path=None,
sample_num=-1,
with_background=True,
use_default_label=None,
**kwargs):
super(ImageFolder, self).__init__(image_dir, anno_path, sample_num,
dataset_dir, with_background)
self.anno_path = anno_path
self.sample_num = sample_num
self.with_background = with_background
super(ImageFolder, self).__init__(dataset_dir, image_dir, anno_path,
sample_num, with_background,
use_default_label)
self.roidbs = None
self._imid2path = {}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册