035、imgaug图像增强之keypoint变换

东方耀

035、imgaug图像增强之keypoint变换


imgaug支持在图像变换的同时变换图像中的关键点

1. # -*- coding: utf-8 -*-
   
2. __author__ = u'东方耀 微信：dfy_88888'
   
3. __date__ = '2020/1/13 16:21'
   
4. __product__ = 'PyCharm'
   
5. __filename__ = '02_keypoint变换'
   
6. 
   
7. import imgaug as ia
   
8. import imgaug.augmenters as iaa
   
9. import numpy as np
   
10. import imageio
    
11. 
    
12. print(ia.__version__)
    
13. # imgaug支持在图像变换的同时变换图像中的关键点
    
14. 
    
15. img_path = 'dfy_imgs/ai111.jpg'
    
16. img = imageio.imread(img_path)
    
17. print(type(img), img.shape)
    
18. ia.seed(666)
    
19. 
    
20. # 定义4个关键点
    
21. keypoints = ia.KeypointsOnImage([
    
22.     ia.Keypoint(x=65, y=100),
    
23.     ia.Keypoint(x=75, y=200),
    
24.     ia.Keypoint(x=100, y=100),
    
25.     ia.Keypoint(x=200, y=80)
    
26. ], shape=img.shape)
    
27. print(type(keypoints), keypoints.shape)
    
28. 
    
29. # 定义一个变换序列
    
30. seq = iaa.Sequential([
    
31.     iaa.Multiply(mul=(1.2, 1.5)),   # 改变亮度,不影响关键点
    
32.     iaa.Affine(rotate=20, scale=0.8)   # 旋转10度然后缩放,会影响关键点
    
33. ])
    
34. 
    
35. # 固定变换序列,之后就可以先变换图像然后变换关键点,这样可以保证两次的变换完全相同。
    
36. # 如果调用次函数,需要在每次batch的时候都调用一次,否则不同的batch执行相同的变换。
    
37. # deterministic 确定性的
    
38. seq_det = seq.to_deterministic()
    
39. 
    
40. # 转换成list或者batch来变换。由于只有一张图片, 因此用[0]来取出该图和关键点。
    
41. image_aug = seq_det.augment_images([img])[0]
    
42. keypoints_aug = seq_det.augment_keypoints([keypoints])[0]
    
43. 
    
44. # print coordinates before/after augmentation (打印图像增强前后的关键点坐标)
    
45. # use after.x_int and after.y_int to get rounded integer coordinates
    
46. for i in range(len(keypoints.keypoints)):
    
47.     before = keypoints.keypoints[i]
    
48.     after = keypoints_aug.keypoints[i]
    
49.     print("Keypoint %d: (%.8f, %.8f) -> (%.8f, %.8f)" % (
    
50.         i, before.x, before.y, after.x, after.y)
    
51.     )
    
52. 
    
53. # 将关键点画在图片上。 size是关键点的大小
    
54. # image with keypoints before/after augmentation (shown below)
    
55. image_before = keypoints.draw_on_image(img, size=15)
    
56. image_after = keypoints_aug.draw_on_image(image_aug, size=7)
    
57. 
    
58. ia.imshow(np.hstack([image_before, image_after]))
    
59. 
    
60. 
    

复制代码

xsoft

谢谢老师答疑解惑