|
037、imgaug图像增强之平移bounding box
- import imgaug as ia
- import imgaug.augmenters as iaa
- import numpy as np
- import imageio
- print(ia.__version__)
- img_path = 'dfy_imgs/ai111.jpg'
- image = imageio.imread(img_path)
- print(type(image), image.shape)
- ia.seed(666)
- bbs = ia.BoundingBoxesOnImage([
- ia.BoundingBox(x1=25, y1=25, x2=75, y2=75),
- ia.BoundingBox(x1=100, y1=25, x2=300, y2=75)
- ], shape=image.shape)
- # 两个box先向右移动15像素,然后第2个box再向下移动110像素
- bbs_shifted = bbs.shift(left=15)
- bbs_shifted.bounding_boxes[1] = bbs_shifted.bounding_boxes[1].shift(top=110)
- # Draw images before/after moving BBs
- image = bbs.draw_on_image(image, color=[0, 255, 0], size=2)
- image_after = bbs_shifted.draw_on_image(image, color=[0, 0, 255], size=2)
- ia.imshow(np.hstack([image, image_after]))
复制代码
|
|