python在图片上画特定大小角度矩形框
2017-10-23 14:57
183 查看
做图像识别的时候需要在图片中画出特定大小和角度的矩形框,自己写了一个函数,给定的输入是图片名称,矩形框的位置坐标,长宽和角度,直接输出画好矩形框的图片。
主要思想是先根据x,y坐标和长宽得到矩形,然后通过数学计算得到旋转angle角度后的新矩形框的四个顶点位置坐标,再利用draw.line()函数画出来。
效果图
主要思想是先根据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*heig 4000 ht 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")); }
效果图
相关文章推荐
- 基于python压缩gif图片大小。
- powershell + python 批量更改图片大小
- [python爬虫]如何爬取特定网页的图片
- python3---取图片中特定的像素替换指定的颜色像素
- Python每日练习 04 你有一个目录,装有很多图片,现在你要把它们的尺寸变成不都大于Iphone5分辨率的大小
- Python遍历文件夹,删除空文件夹,过滤特定大小和格式的文件
- 1.Selenium2 自动化测试实战-基于Python语言-设置窗口大小、调用JS调整滚动条、截取图片
- Python实现更改图片尺寸大小的方法(基于Pillow包)
- python中图片处理之调整图片大小
- python读取图片并修改格式与大小的方法
- python+opencv图片旋转矩形分割
- Python OpenCV 图片反色、调整大小、遍历、导出txt
- Python 获取图片文件大小并转换为base64编码
- Python的Tornado框架实现图片上传及图片大小修改功能
- Python---对html文件内容进行搜索取出特定URL地址字符串,保存成列表,并使用每个url下载图片,并保存到硬盘上,使用正则re
- opencv for python (6) 改变一幅图的特定区域 (往一幅图片上加标志)
- Python 图形处理库PIL批量修改图片大小和格式
- 使用python进行图像处理-调整图片大小
- python自带的PIL库扩展图片大小给图片加上文字描述
- Python脚本遍历FastDfs所有目录筛选出所需大小的图片