037、imgaug图像增强之平移bounding box

东方耀

037、imgaug图像增强之平移bounding box

1. 
   
2. import imgaug as ia
   
3. import imgaug.augmenters as iaa
   
4. import numpy as np
   
5. import imageio
   
6. 
   
7. print(ia.__version__)
   
8. 
   
9. img_path = 'dfy_imgs/ai111.jpg'
   
10. image = imageio.imread(img_path)
    
11. print(type(image), image.shape)
    
12. ia.seed(666)
    
13. 
    
14. 
    
15. bbs = ia.BoundingBoxesOnImage([
    
16.     ia.BoundingBox(x1=25, y1=25, x2=75, y2=75),
    
17.     ia.BoundingBox(x1=100, y1=25, x2=300, y2=75)
    
18. ], shape=image.shape)
    
19. 
    
20. # 两个box先向右移动15像素,然后第2个box再向下移动110像素
    
21. bbs_shifted = bbs.shift(left=15)
    
22. bbs_shifted.bounding_boxes[1] = bbs_shifted.bounding_boxes[1].shift(top=110)
    
23. 
    
24. # Draw images before/after moving BBs
    
25. image = bbs.draw_on_image(image, color=[0, 255, 0], size=2)
    
26. image_after = bbs_shifted.draw_on_image(image, color=[0, 0, 255], size=2)
    
27. 
    
28. ia.imshow(np.hstack([image, image_after]))
    

复制代码