From 684fe1b662bec75f161d08f16696573fb7e6ec31 Mon Sep 17 00:00:00 2001 From: wangguanzhong Date: Tue, 31 Dec 2019 16:31:06 +0800 Subject: [PATCH] support use_default_label for infer (#151) --- ppdet/data/source/dataset.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/ppdet/data/source/dataset.py b/ppdet/data/source/dataset.py index 806dd2b12..9cd7be00d 100644 --- a/ppdet/data/source/dataset.py +++ b/ppdet/data/source/dataset.py @@ -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 = {} -- GitLab