python实现图像随机裁剪的示例代码
作者:我坚信阳光灿烂 时间:2021-06-07 07:02:57
实验条件:
从1张图像随机裁剪100张图像
裁剪出图像的大小为 60 x 60
IoU 大于等于 th=0.6 的裁剪框用红色标出,其它裁剪框用蓝色标出
IoU 比对原始区域用绿框标出
实验代码:
import cv2 as cv
import numpy as np
np.random.seed(0)
# get IoU overlap ratio
def iou(a, b):
# get area of a
area_a = (a[2] - a[0]) * (a[3] - a[1])
# get area of b
area_b = (b[2] - b[0]) * (b[3] - b[1])
# get left top x of IoU
iou_x1 = np.maximum(a[0], b[0])
# get left top y of IoU
iou_y1 = np.maximum(a[1], b[1])
# get right bottom of IoU
iou_x2 = np.minimum(a[2], b[2])
# get right bottom of IoU
iou_y2 = np.minimum(a[3], b[3])
# get width of IoU
iou_w = iou_x2 - iou_x1
# get height of IoU
iou_h = iou_y2 - iou_y1
# get area of IoU
area_iou = iou_w * iou_h
# get overlap ratio between IoU and all area
iou = area_iou / (area_a + area_b - area_iou)
return iou
# crop and create database
def crop_bbox(img, gt, Crop_N=200, L=60, th=0.5):
# get shape
H, W, C = img.shape
# each crop
for i in range(Crop_N):
# get left top x of crop bounding box
x1 = np.random.randint(W - L)
# get left top y of crop bounding box
y1 = np.random.randint(H - L)
# get right bottom x of crop bounding box
x2 = x1 + L
# get right bottom y of crop bounding box
y2 = y1 + L
# crop bounding box
crop = np.array((x1, y1, x2, y2))
# get IoU between crop box and gt
_iou = iou(gt, crop)
# assign label
if _iou >= th:
cv.rectangle(img, (x1, y1), (x2, y2), (0,0,255), 1)
label = 1
else:
cv.rectangle(img, (x1, y1), (x2, y2), (255,0,0), 1)
label = 0
return img
# read image
img = cv.imread("../xiyi.jpg")
img1 = img.copy()
# gt bounding box
gt = np.array((87, 51, 169, 113), dtype=np.float32)
# get crop bounding box
img = crop_bbox(img, gt, Crop_N=100, L=60, th=0.6)
# draw gt
cv.rectangle(img, (gt[0], gt[1]), (gt[2], gt[3]), (0,255,0), 1)
cv.rectangle(img1,(gt[0], gt[1]), (gt[2], gt[3]), (0,255,0), 1)
cv.imshow("result1",img1)
cv.imshow("result", img)
cv.imwrite("out.jpg", img)
cv.waitKey(0)
cv.destroyAllWindows()
实验结果:
来源:https://www.cnblogs.com/wojianxin/p/12581240.html
标签:python,图像,裁剪
0
投稿
猜你喜欢
用Python编写个解释器实现方法接受
2023-01-11 15:50:08
mysql proxy问题
2009-02-27 15:54:00
Django 静态文件配置过程详解
2021-07-08 07:27:53
python中的五种异常处理机制介绍
2022-05-05 13:02:01
python实现录制全屏和选择区域录屏功能
2022-06-18 01:06:33
Django + Uwsgi + Nginx 实现生产环境部署的方法
2023-01-01 06:02:49
python垃圾回收机制(GC)原理解析
2023-01-05 10:57:46
Python3 pyecharts生成Html文件柱状图及折线图代码实例
2022-03-17 01:01:46
从HTTP状态 301,302,200 来看页面跳转
2007-09-26 13:46:00
PHP GD 图像处理组件的常用函数总结
2023-11-21 21:39:17
Tensorflow使用支持向量机拟合线性回归
2021-01-20 03:38:42
Linux安装Python3如何和系统自带的Python2并存
2023-08-25 03:42:09
link 和 style 元素在 HTML 文档中的位置
2008-06-02 13:56:00
Django显示可视化图表的实践
2023-04-13 02:42:38
Python 高级库15 个让新手爱不释手(推荐)
2023-06-13 18:11:03
Python之多进程与多线程的使用
2021-01-10 15:03:57
Python可变集合和不可变集合的构造方法大全
2021-10-23 12:39:04
Python扫描IP段查看指定端口是否开放的方法
2023-10-14 22:57:17
Python微信公众号开发平台
2021-11-25 20:50:12
PHP中error_reporting()函数的用法(修改PHP屏蔽错误)
2023-11-20 01:08:17