當前位置: 首頁>>代碼示例>>Python>>正文


Python boxes.clip_xyxy_to_image方法代碼示例

本文整理匯總了Python中utils.boxes.clip_xyxy_to_image方法的典型用法代碼示例。如果您正苦於以下問題:Python boxes.clip_xyxy_to_image方法的具體用法?Python boxes.clip_xyxy_to_image怎麽用?Python boxes.clip_xyxy_to_image使用的例子?那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在utils.boxes的用法示例。


在下文中一共展示了boxes.clip_xyxy_to_image方法的1個代碼示例,這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚,您的評價將有助於係統推薦出更棒的Python代碼示例。

示例1: _add_gt_annotations

# 需要導入模塊: from utils import boxes [as 別名]
# 或者: from utils.boxes import clip_xyxy_to_image [as 別名]
def _add_gt_annotations(self, entry):
        """Add ground truth annotation metadata to an roidb entry."""
        ann_ids = self.COCO.getAnnIds(imgIds=entry['id'], iscrowd=None)
        objs = self.COCO.loadAnns(ann_ids)
        # Sanitize bboxes -- some are invalid
        valid_objs = []
        valid_segms = []
        width = entry['width']
        height = entry['height']
        for obj in objs:
            # crowd regions are RLE encoded and stored as dicts
            if obj['area'] < cfg.TRAIN.GT_MIN_AREA:
                continue
            if 'ignore' in obj and obj['ignore'] == 1:
                continue
            # Convert form (x1, y1, w, h) to (x1, y1, x2, y2)
            x1, y1, x2, y2 = box_utils.xywh_to_xyxy(obj['bbox'])
            x1, y1, x2, y2 = box_utils.clip_xyxy_to_image(
                x1, y1, x2, y2, height, width
            )
            # Require non-zero seg area and more than 1x1 box size
            if obj['area'] > 0 and x2 > x1 and y2 > y1:
                obj['clean_bbox'] = [x1, y1, x2, y2]
                valid_objs.append(obj)
        num_valid_objs = len(valid_objs)

        gt_classes = np.zeros((num_valid_objs), dtype=entry['gt_classes'].dtype)
        for ix, obj in enumerate(valid_objs):
            cls = self.json_category_id_to_contiguous_id[obj['category_id']]
            gt_classes[ix] = cls

        for cls in gt_classes:
            entry['gt_classes'][0, cls] = 1 
開發者ID:ppengtang,項目名稱:pcl.pytorch,代碼行數:35,代碼來源:json_dataset.py


注:本文中的utils.boxes.clip_xyxy_to_image方法示例由純淨天空整理自Github/MSDocs等開源代碼及文檔管理平台,相關代碼片段篩選自各路編程大神貢獻的開源項目,源碼版權歸原作者所有,傳播和使用請參考對應項目的License;未經允許,請勿轉載。