diff --git a/paddlehub/dataset/base_cv_dataset.py b/paddlehub/dataset/base_cv_dataset.py index 7e0b687d1d93ff91ee15f51a8218e6266a8147e2..2da367722a8ec1ef9c88d37946f470cb62d77268 100644 --- a/paddlehub/dataset/base_cv_dataset.py +++ b/paddlehub/dataset/base_cv_dataset.py @@ -52,7 +52,13 @@ class ImageClassificationDataset(object): break line = line.strip() items = line.split(" ") - image_path = os.path.join(self.base_path, items[0]) + if os.path.isabs(items[0]): + image_path = items[0] + else: + if self.base_path is None: + image_path = items[0] + else: + image_path = os.path.join(self.base_path, items[0]) label = items[1] data.append((image_path, items[1]))