用numpy/torch旋转图片的内圆

原创已于 2024-10-30 10:40:24 修改 · 3.3k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #算法 #图像处理

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

本文介绍如何利用numpy和torch库来实现图像中圆形区域的旋转操作，通过提供的代码示例详细展示了具体步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

活不多说，先上图

img = cv2.imread('OIP-C.jpeg',cv2.IMREAD_COLOR)
imgRGB = cv2.cvtColor(img,cv2.COLOR_BGR2RGB)
dst = cv2.resize(imgRGB, (400, 400))
Image.fromarray(dst)

 Image.fromarray(rotating_inscribed_circle(dst,np.pi*2/3,120))

方法代码：

import numpy as np
def rotating_inscribed_circle(img,radian,r=None):
    #img : (h,w,3)
    h,w,_=img.shape
    r=min(w,h)/2 if r is None else r
    y=np.arange(h).reshape(-1,1)
    x=np.arange(w).reshape(1,-1)
    cy ,cx= (h-1)/2,(w-1)/2
    x_=(y-cy)*np.sin(radian)+(x-cx)*np.cos(radian)+cx
    y_=(y-cy)*np.cos(radian)-(x-cx)*np.sin(radian)+cy
    i = (y_-cy)**2+(x_-cx)**2 <r**2
    y=(y*(1-i)+y_*i).astype(int)
    x=(x*(1-i)+x_*i).astype(int)
    return img[y,x]