From 13501bfb6f97b8b0aa0876afd1b9adc9535e0c3f Mon Sep 17 00:00:00 2001 From: sunyanfang01 Date: Fri, 8 May 2020 21:08:55 +0800 Subject: [PATCH] fix the bug --- paddlex/cv/datasets/easydata_cls.py | 2 +- paddlex/cv/datasets/easydata_det.py | 2 +- paddlex/cv/datasets/easydata_seg.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/paddlex/cv/datasets/easydata_cls.py b/paddlex/cv/datasets/easydata_cls.py index ecf0f75..c947206 100644 --- a/paddlex/cv/datasets/easydata_cls.py +++ b/paddlex/cv/datasets/easydata_cls.py @@ -76,7 +76,7 @@ class EasyDataCls(ImageNet): raise IOError( 'The image file {} is not exist!'.format(img_file)) with open(json_file, mode='r', \ - encoding=get_encoding(label_list)) as j: + encoding=get_encoding(json_file)) as j: json_info = json.load(j) label = json_info['labels'][0]['name'] self.file_list.append([img_file, self.labels.index(label)]) diff --git a/paddlex/cv/datasets/easydata_det.py b/paddlex/cv/datasets/easydata_det.py index d22cb39..87f5bac 100644 --- a/paddlex/cv/datasets/easydata_det.py +++ b/paddlex/cv/datasets/easydata_det.py @@ -96,7 +96,7 @@ class EasyDataDet(VOCDetection): raise IOError( 'The image file {} is not exist!'.format(img_file)) with open(json_file, mode='r', \ - encoding=get_encoding(label_list)) as j: + encoding=get_encoding(json_file)) as j: json_info = json.load(j) im_id = np.array([ct]) im = cv2.imread(img_file) diff --git a/paddlex/cv/datasets/easydata_seg.py b/paddlex/cv/datasets/easydata_seg.py index 7828524..4e5f511 100644 --- a/paddlex/cv/datasets/easydata_seg.py +++ b/paddlex/cv/datasets/easydata_seg.py @@ -79,7 +79,7 @@ class EasyDataSeg(Dataset): raise IOError( 'The image file {} is not exist!'.format(img_file)) with open(json_file, mode='r', \ - encoding=get_encoding(label_list)) as j: + encoding=get_encoding(json_file)) as j: json_info = json.load(j) im = cv2.imread(img_file) im_w = im.shape[1] -- GitLab