diff --git a/paddlex/cv/datasets/easydata_det.py b/paddlex/cv/datasets/easydata_det.py index 87f5bac4d0b40844621b58bb824432846c22972c..6424806148d970e8ee868eef089f7b025c333bd7 100644 --- a/paddlex/cv/datasets/easydata_det.py +++ b/paddlex/cv/datasets/easydata_det.py @@ -19,8 +19,6 @@ import copy import json import cv2 import numpy as np -from pycocotools.coco import COCO -from pycocotools.mask import decode import paddlex.utils.logging as logging from .voc import VOCDetection from .dataset import is_pic @@ -82,6 +80,7 @@ class EasyDataDet(VOCDetection): 'name': k }) + from pycocotools.mask import decode ct = 0 ann_ct = 0 with open(file_list, encoding=get_encoding(file_list)) as f: @@ -133,7 +132,7 @@ class EasyDataDet(VOCDetection): 'area': float((x2 - x1 + 1) * (y2 - y1 + 1)), 'segmentation': - [] if gt_poly[i] is None else gt_poly[i], + [[x1, y1, x1, y2, x2, y2, x2, y1]] if gt_poly[i] is None else gt_poly[i], 'category_id': cname2cid[cname], 'id': @@ -175,6 +174,7 @@ class EasyDataDet(VOCDetection): logging.info("{} samples in file {}".format( len(self.file_list), file_list)) self.num_samples = len(self.file_list) + from pycocotools.coco import COCO self.coco_gt = COCO() self.coco_gt.dataset = annotations self.coco_gt.createIndex() diff --git a/paddlex/cv/datasets/easydata_seg.py b/paddlex/cv/datasets/easydata_seg.py index 4e5f51135ad9ff24c0a6a4433b67c22032040bf3..56f801721d435e9c0c92809c2aac35f43c074b7e 100644 --- a/paddlex/cv/datasets/easydata_seg.py +++ b/paddlex/cv/datasets/easydata_seg.py @@ -19,7 +19,6 @@ import copy import json import cv2 import numpy as np -from pycocotools.mask import decode import paddlex.utils.logging as logging from .dataset import Dataset from .dataset import get_encoding @@ -59,6 +58,7 @@ class EasyDataSeg(Dataset): self.labels = list() self._epoch = 0 + from pycocotools.mask import decode cname2cid = {} label_id = 0 with open(label_list, encoding=get_encoding(label_list)) as fr: