?
题目:给出旋转矩形的四个顶点 p1(x1, y1), p2(x2, y2), p3(x3, y3), p4(x4, y4) 或者 中心点 (cx,cy),(w,h),θ 画一个旋转的实心矩阵:输出numpy数组,矩形内的值全部为255
一、通过中心点、宽高、角度得到四个顶点的公式网上都有,最后输出四个顶点的x和y坐标。
import numpy as np
rectangle = [cx,cy,w,h,θ]
x1 = int(np.cos(rectangle[4])*(-rectangle[2]/2) - np.sin(rectangle[4])*(-rectangle[3]/2) + rectangle[0])
x2 = int(np.cos(rectangle[4])*(rectangle[2]/2) - np.sin(rectangle[4])*(-rectangle[3]/2) + rectangle[0])
x3 = int(np.cos(rectangle[4])*(-rectangle[2]/2) - np.sin(rectangle[4])*(rectangle[3]/2) + rectangle[0])
x4 = int(np.cos(rectangle[4])*(rectangle[2]/2) - np.sin(rectangle[4])*(rectangle[3]/2) + rectangle[0])
y1 = int(np.sin(rectangle[4])*(-rectangle[2]/2) + np.cos(rectangle[4])*(-rectangle[3]/2) + rectangle[1])
y2 = int(np.sin(rectangle[4])*(rectangle[2]/2) + np.cos(rectangle[4])*(-rectangle[3]/2) + rectangle[1])
y3 = int(np.sin(rectangle[4])*(-rectangle[2]/2) + np.cos(rectangle[4])*(rectangle[3]/2) + rectangle[1])
y4 = int(np.sin(rectangle[4])*(rectangle[2]/2) + np.cos(rectangle[4])*(rectangle[3]/2) + rectangle[1])
二、绘制实心矩形
关于怎么绘制实心矩形,因为openCV本身的cv2.rectangle只能绘制角度为0的矩形,没办法用,所以一开始打算遍历每个点判断是否在矩形内,来绘制
from shapely import geometry
def ifPointsInside(polygon, Points):
line = geometry.LineString(polygon)
point = geometry.Point(Points)
polygon = geometry.Polygon(line)
return polygon.contains(point)
以及用凹凸多边形来判断、用四个三角形面积和矩形面积对比来找
结果因为每个点都要遍历计算,我是用来处理深度学习资料的,一万多张图片,640*480分辨率,那就是30多亿次,他没遍历完,我人就先走了,太慢
然后想到了:先用cv2.line绘制边框,然后图像按横遍历找里面的255值,找到两个255的话就把这两个中间的全都标记255,成功了,不过还是慢
最后解决方法:用cv2.line一条线一条线的画,从(x1,y1)->(x2,y2)一直画到(x3,y3)->(x4,y4)
Output = np.zeros((480,640,1))
a = (y3-y1)/(x3-x1)
for x in range(0,abs(x3-x1)):
cv2.line(Output,(int(x1+x),int(y1+a*x)),(int(x2+x),int(y2+a*x)), (255), 2)
按1pixel画,精度不够,改为0.01步伐
Output = np.zeros((480,640,1))
a = (y3-y1)/(x3-x1)
for x in range(0,100*abs(x3-x1)):
cv2.line(Output,(int(x1+x/100),int(y1+a*x/100)),(int(x2+x/100),int(y2+a*x/100)), (255), 2)
画出来了
|