人工智能视频教程 ai vip技术 人工智能数学基础 爬虫 python机器学习 tensorflow深度学习 20+个企业AI实战项目

 找回密码
 立即注册

QQ登录

只需一步,快速开始

搜索
热搜: 活动 交友 discuz
查看: 136|回复: 0

[学习笔记] 037、imgaug图像增强之平移bounding box

[复制链接]

1072

主题

1395

帖子

1万

积分

管理员

Rank: 10Rank: 10Rank: 10

积分
11716
QQ
发表于 2020-1-13 20:03:49 | 显示全部楼层 |阅读模式


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


  1. import imgaug as ia
  2. import imgaug.augmenters as iaa
  3. import numpy as np
  4. import imageio

  5. print(ia.__version__)

  6. img_path = 'dfy_imgs/ai111.jpg'
  7. image = imageio.imread(img_path)
  8. print(type(image), image.shape)
  9. ia.seed(666)


  10. bbs = ia.BoundingBoxesOnImage([
  11.     ia.BoundingBox(x1=25, y1=25, x2=75, y2=75),
  12.     ia.BoundingBox(x1=100, y1=25, x2=300, y2=75)
  13. ], shape=image.shape)

  14. # 两个box先向右移动15像素,然后第2个box再向下移动110像素
  15. bbs_shifted = bbs.shift(left=15)
  16. bbs_shifted.bounding_boxes[1] = bbs_shifted.bounding_boxes[1].shift(top=110)

  17. # Draw images before/after moving BBs
  18. image = bbs.draw_on_image(image, color=[0, 255, 0], size=2)
  19. image_after = bbs_shifted.draw_on_image(image, color=[0, 0, 255], size=2)

  20. ia.imshow(np.hstack([image, image_after]))
复制代码


平移bounding box.jpg
让天下人人学会人工智能!人工智能的前景一片大好!
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

QQ|Archiver|手机版|小黑屋|人工智能工程师的摇篮 ( 湘ICP备18018285号-1 )

GMT+8, 2020-8-4 01:39 , Processed in 0.227279 second(s), 22 queries .

Powered by Discuz! X3.4

© 2001-2017 Comsenz Inc.

快速回复 返回顶部 返回列表