python实现在图片上画特定大小角度矩形框
作者:Imcy 发布时间:2023-03-16 22:03:14
标签:python,矩形框
做图像识别的时候需要在图片中画出特定大小和角度的矩形框,自己写了一个函数,给定的输入是图片名称,矩形框的位置坐标,长宽和角度,直接输出画好矩形框的图片。
主要思想是先根据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
0
投稿
猜你喜欢
- 功能描述:打开一个网站的网页,过5分钟不动作,就会锁定页面,隐藏内容容器,显示一个容器用于输入密码,输入正确的密码来解锁。锁定后即使用户刷新
- 1. 打开FrontPage 2003,点击“文件→新建→新建网站→其他网站模板”,然后选择“数据库界面向导”,给定网站路径后,单击[确定]
- 这篇文章主要介绍了微信小程序顶部导航栏可滑动并选中放大,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋
- 1. Document.form.item 问题 现有问题:  
- 用户体验已经是一个老生常谈的话题了。我非常赞同某位达人所说的,用户体验设计应该贯穿于产品从萌芽到出生的整个过程,产品原型、视觉设计、前端开发
- 内容摘要:MySQL易学易用,附带丰富的技术文档,这两个因素使之被广泛应用。然而,随着MySQL发展加快,即使一个MySQL老手有时也会为该
- 1. 用Dreamweaver 4.0制作闪动的Flash按钮选择菜单Insert→Interactive Images→Flash But
- 星爷的一部“国产007”应该是无人不晓,其中一个片段是将“007”向MM展示他的秘密武器。皮鞋可以吹头发,大哥大可以刮胡子……把真实功能隐蔽
- 代码如下:---1.平均销售等待时间 ---有一张Sales表,其中有销售日期与顾客两列,现在要求使用一条SQL语句实现计算 --每个顾客的
- 我们用下了asp代码简单统计了下载一个文件需要的时间:<%Function DownloadTime(intFileSize
- 天我们看看import的有关内容。编程时总是用到import导入,动不动就导入,很简单,但import到底是个什么功能,它的本质是什么?一.
- 原文地址:30 Days of Mootools 1.2 Tutorials - Day 7 - Set and Get Style Pro
- 一、什么是Oracle字符集 Oracle字符集是一个字节数据的解释的符号集合,有大小之分,有相互的包容关系。ORACLE 支持国家语言的体
- 经过一个阶段的asp学习,下面我们结合所学过的内容建立一个最简单的聊天室,虽然很简单,但是大家可以通过他来掌握一个聊天室建立的基本过程,并且
- 数独规则数独游戏,经典的为9×9=81个单元格组成的九宫格,同时也形成了3×3=9个小九宫格,要求在81个小单元格中填入数字1~9,并且数字
- ISSET();——适合于检测是否存在这个参数。 定义和作用范围:用于测试一个变量是否具有值(包括0,FALSE,或者一个空字串,但不能是N
- 十要:第一:要认真规划和分析。这是网页设计灵魂工作。创建站点之前,要明确你的网站主要针对哪些访问者,为哪些用户服务,要把握准主页题材第二:网
- PHP有一组进程控制函数(编译时需要–enable-pcntl与posix扩展),使得php能实现跟c
- 安装环境:python版本2.7.5 ,win7系统安装Djangohttps://www.djangoproject.com/downlo
- 本文实例讲述了php遍历CSV类。分享给大家供大家参考。具体如下:<?phpclass CSVIterator implements