python实现在图片上画特定大小角度矩形框
作者:Imcy 时间:2023-03-16 22:03:14
做图像识别的时候需要在图片中画出特定大小和角度的矩形框,自己写了一个函数,给定的输入是图片名称,矩形框的位置坐标,长宽和角度,直接输出画好矩形框的图片。
主要思想是先根据x,y坐标和长宽得到矩形,然后通过数学计算得到旋转angle角度后的新矩形框的四个顶点位置坐标,再利用draw.line()函数画出来。
import math
import matplotlib.pyplot as plt
import numpy as np
from PIL import Image, ImageDraw
def draw(filename,result):
img = Image.open(filename)
w,h=img.size
draw = ImageDraw.Draw(img)
result=np.array(result)
x=result[0][0]
y=result[0][1]
angle=result[0][2]
height=result[0][3]
width=result[0][4]
anglePi = -angle*math.pi/180.0
cosA = math.cos(anglePi)
sinA = math.sin(anglePi)
x1=x-0.5*width
y1=y-0.5*height
x0=x+0.5*width
y0=y1
x2=x1
y2=y+0.5*height
x3=x0
y3=y2
x0n= (x0 -x)*cosA -(y0 - y)*sinA + x
y0n = (x0-x)*sinA + (y0 - y)*cosA + y
x1n= (x1 -x)*cosA -(y1 - y)*sinA + x
y1n = (x1-x)*sinA + (y1 - y)*cosA + y
x2n= (x2 -x)*cosA -(y2 - y)*sinA + x
y2n = (x2-x)*sinA + (y2 - y)*cosA + y
x3n= (x3 -x)*cosA -(y3 - y)*sinA + x
y3n = (x3-x)*sinA + (y3 - y)*cosA + y
draw.line([(x0n, y0n),(x1n, y1n)], fill=(0, 0, 255))
draw.line([(x1n, y1n),(x2n, y2n)], fill=(255, 0, 0))
draw.line([(x2n, y2n),(x3n, y3n)],fill= (0,0,255))
draw.line([(x0n, y0n), (x3n, y3n)],fill=(255,0,0))
plt.imshow(img)
plt.show()
改写成c++
由于c++程序也需要用到这个程序,改写一下
void CopentestDlg::OnBnClickedAnaly()
{
CString x,y,angle,width;
float xRaw[4], yRaw[4];
float xNew[4], yNew[4];
float height = 33;
if (imageCut==NULL) {
AfxMessageBox(_T("未打开图片不能分析"));
return;
}
GetDlgItem(IDC_EX)->GetWindowText(x);
GetDlgItem(IDC_EY)->GetWindowText(y);
GetDlgItem(IDC_ANGLE)->GetWindowText(angle);
GetDlgItem(IDC_WIDTH)->GetWindowText(width);
float xf = _tstof(x); //将Cstring转成float类型
float yf = _tstof(y);
float widthf= _tstof(width);
float anglef= _tstof(angle);
xRaw[1] = xf - 0.5*widthf;
yRaw[1] = yf - 0.5*height;
xRaw[0] = xf + 0.5*widthf;
yRaw[0] = yRaw[1];
xRaw[2] = xRaw[1];
yRaw[2] = yf + 0.5*height;
xRaw[3] = xRaw[0];
yRaw[3] = yRaw[2];
float anglePi = -anglef*pi / 180.0;
float cosA = cos(anglePi);
float sinA = sin(anglePi);
xNew[0] = (xRaw[0] - xf)*cosA - (yRaw[0] - yf)*sinA + xf;
yNew[0] = (xRaw[0] - xf)*sinA + (yRaw[0] - yf)*cosA + yf;
xNew[1] = (xRaw[1] - xf)*cosA - (yRaw[1] - yf)*sinA + xf;
yNew[1] = (xRaw[1] - xf)*sinA + (yRaw[1] - yf)*cosA + yf;
xNew[2] = (xRaw[2] - xf)*cosA - (yRaw[2] - yf)*sinA + xf;
yNew[2] = (xRaw[2] - xf)*sinA + (yRaw[2] - yf)*cosA + yf;
xNew[3] = (xRaw[3] - xf)*cosA - (yRaw[3] - yf)*sinA + xf;
yNew[3] = (xRaw[3] - xf)*sinA + (yRaw[3] - yf)*cosA + yf;
cvLine(imageCut, cvPoint(xNew[0], yNew[0]), cvPoint(xNew[1], yNew[1]), cvScalar(0, 0, 255), 3);
cvLine(imageCut, cvPoint(xNew[1], yNew[1]), cvPoint(xNew[2], yNew[2]), cvScalar(255, 0, 0), 3);
cvLine(imageCut, cvPoint(xNew[2], yNew[2]), cvPoint(xNew[3], yNew[3]), cvScalar(0, 0, 255), 3);
cvLine(imageCut, cvPoint(xNew[0], yNew[0]), cvPoint(xNew[3], yNew[3]), cvScalar(255, 0, 0), 3);
DrawPicToHDC(imageCut, IDC_Cut);
pEdit->ReplaceSel(_T("图片已修改\r\n"));
}
效果图
来源:https://blog.csdn.net/u013948010/article/details/78318894
标签:python,矩形框
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
flask实现验证码并验证功能
2022-02-03 07:40:02
![](https://img.aspxhome.com/file/2023/8/133708_0s.jpg)
python的Template使用指南
2023-10-14 01:30:58
Python图片视频超分模型RealBasicVSR的使用教程
2021-10-11 13:40:04
![](https://img.aspxhome.com/file/2023/4/105244_0s.jpg)
python框架Django实战商城项目之工程搭建过程图文详解
2022-12-16 16:25:57
![](https://img.aspxhome.com/file/2023/6/72046_0s.jpg)
JS变量中有var定义和无var定义的区别以及es6中let命令和const命令
2024-05-22 10:37:29
![](https://img.aspxhome.com/file/2023/0/123410_0s.jpg)
陌生网页交互行为分析(1)——奇怪的关闭按钮
2009-01-08 12:22:00
![](https://img.aspxhome.com/file/UploadPic/20091/8/2008-12-30_155447-20s.png)
Golang 处理浮点数遇到的精度问题(使用decimal)
2024-04-25 15:01:15
![](https://img.aspxhome.com/file/2023/1/134241_0s.png)
asp随机获取access数据库中的一条记录
2007-08-15 13:11:00
Python Pyperclip模块安装和使用详解
2023-06-25 13:52:11
Python浅析迭代器Iterator的使用
2023-11-07 12:04:25
python用pip install时安装失败的一系列问题及解决方法
2021-10-15 23:11:43
![](https://img.aspxhome.com/file/2023/2/120412_0s.png)
关于反爬虫的一些简单总结
2022-09-21 18:38:59
![](https://img.aspxhome.com/file/2023/7/109537_0s.jpg)
Ajax学习小贴士
2007-10-24 23:21:00
python DataFrame 修改列的顺序实例
2023-07-21 12:55:40
vue3和ts封装axios以及使用mock.js详解
2024-04-28 09:27:47
![](https://img.aspxhome.com/file/2023/5/132955_0s.png)
MySQL备份脚本的写法
2024-01-18 01:35:08
PyCharm 光标变成黑块的解决方式
2023-11-12 11:02:48
![](https://img.aspxhome.com/file/2023/6/127726_0s.jpg)
基于Python实现人脸识别和焦点人物检测功能
2021-03-09 10:27:45
![](https://img.aspxhome.com/file/2023/5/107995_0s.jpg)
如何获取文件的名称和扩展名?
2009-11-23 20:50:00
SqlServer应用之sys.dm_os_waiting_tasks 引发的疑问(上)
2024-01-12 15:24:51
![](https://img.aspxhome.com/file/2023/6/88266_0s.png)