东方耀AI技术分享
标题:
037、imgaug图像增强之平移bounding box
[打印本页]
作者:
东方耀
时间:
2020-1-13 20:03
标题:
037、imgaug图像增强之平移bounding box
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]))
复制代码
欢迎光临 东方耀AI技术分享 (http://www.ai111.vip/)
Powered by Discuz! X3.4