当前位置： 移动技术网 > IT编程>脚本编程>Python > python实现在图片上画特定大小角度矩形框

python实现在图片上画特定大小角度矩形框

2018年11月05日  | 移动技术网IT编程  | 我要评论

别人笑我太疯癫全诗,香格里拉藏民家访,星剧社

做图像识别的时候需要在图片中画出特定大小和角度的矩形框，自己写了一个函数，给定的输入是图片名称，矩形框的位置坐标,长宽和角度，直接输出画好矩形框的图片。

主要思想是先根据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"));
}

效果图

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持移动技术网。

您可能感兴趣的文章:

如对本文有疑问，请在下面进行留言讨论，广大热心网友会与你互动！！ 点击进行留言回复

相关文章:

验证码：
最近更新的文章
大家感兴趣的文章
移动技术网