Ejemplo n.º 1
0
 def _parse_annotation(self,itemidx,random_trainsize):
     rootpath, filename = self._ids[itemidx]
     annpath = self._annopath.format(rootpath, filename)
     imgpath = self._imgpath.format(rootpath, filename)
     fname, bboxes, labels = PascalVocXmlParser(annpath, self.labels).parse()
     img = cv2.imread(imgpath, cv2.IMREAD_COLOR)
     if self.istrain:
         img, bboxes = dataAug.random_horizontal_flip(np.copy(img), np.copy(bboxes))
         img, bboxes = dataAug.random_crop(np.copy(img), np.copy(bboxes))
         img, bboxes = dataAug.random_translate(np.copy(img), np.copy(bboxes))
     ori_shape=img.shape[:2]
     img, bboxes = dataAug.img_preprocess2(np.copy(img), np.copy(bboxes),
                                           (random_trainsize, random_trainsize), True)
     return img,bboxes,labels,imgpath,ori_shape
Ejemplo n.º 2
0
 def _parse_annotation(self, itemidx, random_trainsize):
     img_info = self.img_infos[itemidx]
     ann_info = self._load_ann_info(itemidx)
     ann = self._parse_ann_info(ann_info)
     bboxes = ann['bboxes']  # [x1,y1,x2,y2]
     labels = ann['labels']
     # load the image.
     imgpath = osp.join(self.image_dir, img_info['file_name'])
     img = cv2.imread(imgpath, cv2.IMREAD_COLOR)
     if self.istrain:
         img, bboxes = dataAug.random_horizontal_flip(
             np.copy(img), np.copy(bboxes))
         img, bboxes = dataAug.random_crop(np.copy(img), np.copy(bboxes))
         img, bboxes = dataAug.random_translate(np.copy(img),
                                                np.copy(bboxes))
     ori_shape = img.shape[:2]
     img, bboxes = dataAug.img_preprocess2(
         np.copy(img), np.copy(bboxes),
         (random_trainsize, random_trainsize), True)
     return img, bboxes, labels, imgpath, ori_shape