当前位置：首页 > article >正文

同时增强多个目标：masks, bounding boxes, keypoints

article 2024/11/15 5:07:50

参考链接：

Simultaneous augmentation of multiple targets: masks, bounding boxes, keypoints - Albumentations Documentation

Albumentations 可以将相同的一组变换应用于 the input image 和 all the targets 传递到transform：masks, bounding boxes, keypoints。

Step 1. 使用指定边界框和关键点格式的参数定义 Compose 。

transform = A.Compose(
  [A.RandomCrop(width=330, height=330), A.RandomBrightnessContrast(p=0.2)],
  bbox_params=A.BboxParams(format="coco", label_fields=["bbox_classes"]),
  keypoint_params=A.KeypointParams(format="xy", label_fields=["keypoints_classes"]),
)

Step 2. 从磁盘加载所有必需的数据。

For example, here is an image from the COCO dataset. that has one associated mask, one bounding box with the class label person, and five keypoints that define body parts.

An example image with mask, bounding boxes and keypoints

Step 3. 传递所有目标进行转换并接收其增强版本¶

transformed = transform(
  image=img,
  mask=mask,
  bboxes=bboxes,
  bbox_classes=bbox_classes,
  keypoints=keypoints,
  keypoints_classes=keypoints_classes,
)
transformed_image = transformed["image"]
transformed_mask = transformed["mask"]
transformed_bboxes = transformed["bboxes"]
transformed_bbox_classes = transformed["bbox_classes"]
transformed_keypoints = transformed["keypoints"]
transformed_keypoints_classes = transformed["keypoints_classes"]