用numpy/torch将圆环映射成矩形

janelu9

已于 2022-04-29 21:42:14 修改

阅读量3.2k

点赞数

文章标签：图像处理 python

于 2022-04-25 16:04:54 首次发布

本文链接：https://blog.csdn.net/lujian45/article/details/124407431

版权

本文介绍如何使用numpy和torch库将圆环区域的图像映射成矩形。通过示例代码展示转换过程，帮助理解图像处理中的几何变换。

摘要由CSDN通过智能技术生成

话不多说，先上图：

img = cv2.imread('yz.jpeg',cv2.IMREAD_COLOR)
imgRGB = cv2.cvtColor(img,cv2.COLOR_BGR2RGB)
dst = cv2.resize(imgRGB, (400, 400))

Image.fromarray(circle2rectangle(dst,(200,200),200,100))

方法代码：

def circle2rectangle(img,circle_center,R,W):
    C=int(2*R*np.pi)
    rectangle = np.zeros((W,C,3),dtype='uint8')
    x=np.arange(C).reshape(-1,1)
    y=np.arange(W).reshape(1,-1)
    x,y=np.broadcast_arrays(x,y)
    theta = np.pi*2/C*(x+1)
    r= R-y-1
    p_x = (circle_center[0] - r*np.sin(theta)+0.5).astype(int)-1
    p_y = (circle_center[1] + r*np.cos(theta)+0.5).astype(int)-1
    rectangle[y,x,:] = img[p_y,p_x,:]
    return rectangle