基于Python实现椭圆拟合

资源下载地址:https://download.csdn.net/download/sheziqiong/85836974
资源下载地址:https://download.csdn.net/download/sheziqiong/85836974

椭圆拟合

实验目的和要求

尝试使用 cv.fitEllipse()函数,对图像进行椭圆拟合

实验内容和原理

椭圆拟合

该函数使用的是最小二乘法拟合,要求输入的点至少有 6 个。

基于Python实现椭圆拟合

基于Python实现椭圆拟合

基于Python实现椭圆拟合

基于Python实现椭圆拟合

基于Python实现椭圆拟合

基于Python实现椭圆拟合

函数中对应的参数如下:

基于Python实现椭圆拟合

对输入图像的预处理

输入一张 RGB 图片,先转换为灰度图,本来打算先转换为二值图像再进行边缘检测的,但是发现二值化容易使阴影成为新的边缘,并丢失原有边缘信息,于是直接对灰度图进行了边缘检测。在边缘检测前还进行了降噪

边缘检测

用 Sobel Kernel 进行滤波,来计算 x 和 y 方向的导数。

基于Python实现椭圆拟合

对于每个像素,检测它是否为局部最大值,如果不是就设置为 0

基于Python实现椭圆拟合

输入的两个参数为 min 和 max,用来限定寻找的边缘范围(大于 max 的肯定是边缘,小于 min 的肯定不是,在 min 和 max 之间的则通过判断连通性等方式来判断。

实验步骤与分析

读入图片

读入后先去掉高频的信息(降噪)。然后保存一个 RGB 图像信息,存储一个 gray 灰度图信息。边缘信息 edge 直接从 gray 中提取,min 取到 100,max 取到 200。

FILENAME = "dota2items4.bmp"
img = cv.imread(FILENAME)
img = cv.GaussianBlur(img, (5, 5), 0)
rgb = cv.cvtColor(img, cv.COLOR_BGR2RGB)
gray = cv.cvtColor(img, cv.COLOR_BGR2GRAY)

edge = cv.Canny(gray, 100, 200)

使用 fitEllipse()函数

Ells 数组储存椭圆信息,recs 储存可以旋转的最小矩阵,recs2 储存正矩形信息。Cols 用于储存随机生成的颜色信息,方便区分不同椭圆和矩阵之间的对应关系

img = rgb
edge_ = cv.cvtColor(edge, cv.COLOR_GRAY2RGB)
ells = []
recs = []
recs2 = []
cols = []
contours, nothing = cv.findContours(edge, cv.RETR_TREE, cv.CHAIN_APPROX_SIMPLE)
for index, contour in enumerate(contours):
    if contour.shape[0] > 50:
        ells.append(cv.fitEllipse(contour))
        recs2.append(cv.boundingRect(contour))
        recs.append(cv.minAreaRect(contour))

random.seed(10)
for i in range(len(ells)):
    cols.append([random.randint(0, 256) for j in range(3)])

绘制椭圆与矩形

同时在 RGB 图像和边缘图像中绘制,这是椭圆

for index, ell in enumerate(ells):
        cv.ellipse(img, ell, cols[index], 2)
        cv.ellipse(edge_, ell, cols[index], 2)

这是可旋转的矩形

for index, rec in enumerate(recs):
    box = cv.boxPoints(rec) # cv2.boxPoints(rect) for OpenCV 3.x
    box = np.int0(box)
    cv.drawContours(img,[box],0,cols[index],1)
    cv.drawContours(edge_,[box],0,cols[index],1)

绘制正矩形

for index, rec2 in enumerate(recs2):
    x,y,w,h = rec2
    cv.rectangle(img,(x,y),(x+w,y+h),cols[index],1)
    cv.rectangle(edge_,(x,y),(x+w,y+h),cols[index],1)

实验结果

输入图像

图里左上角的 Hyper stone 具有很多同心圆信息,右上的 Clarity 有不规则曲线和直线信息,右下的 Tango 有圆形信息且整体外形近似椭圆,正下方的 Force Staff 有近似的圆形信息和直线。

基于Python实现椭圆拟合

输出:

  • RGB

图像因为进行过高斯滤波而稍微模糊

基于Python实现椭圆拟合
  • GrayScale

基于Python实现椭圆拟合
  • Thresh

可以看出,二值化后的图像失去了很多细节信息,同时引入了阴影的边缘信息,这是我们不太想要的

基于Python实现椭圆拟合
  • Edge

根据灰度图生成了图像的边缘信息

基于Python实现椭圆拟合

Ellipse on Edge

大部分边缘信息拟合良好,但是图中红色箭头标出的浅蓝色椭圆很奇怪,通过分析,它对应的 contour 是左边浅蓝色矩形内部的信息,并不知道为什么会发生这样的事,也许是 bug?(这个现象同样在 Lena 图片中出现,根据搜索到的信息,可能是最小二乘法造成的问题)

基于Python实现椭圆拟合

下图为 Lena 照片下半部分,看得出左边深蓝色矩阵和右边的椭圆对应,但并没有任何重合。

基于Python实现椭圆拟合

Ellipse on RGB

看得出,那个浅蓝色椭圆与整个图片格格不入

基于Python实现椭圆拟合

心得体会

本来想直接用那种波点图像就算了,看见同学做了对奇怪的游戏 CG 的拟合,于是考虑到继续使用 Lena 姐姐的照片,但是羽毛帽的细节信息太多,导致效果并不十分好。于是找了 DOTA2 的物品图标(虽然效果很好,但是分辨率实在太低,于是找了二次创作的高清图像,效果还不错,各种情况下的效果都能看见)

基于Python实现椭圆拟合

比如这里非常标准的椭圆

基于Python实现椭圆拟合

基于Python实现椭圆拟合

这里对形状近似的物体的拟合

基于Python实现椭圆拟合

这个粉色应该是在拟合瓶子的左下边缘。

通过调大拟合时使用的点的数量阈值以及画的线的粗细,可以删除一些不靠谱的结果,防止出现一大团乱七八糟的线 的效果(如下)

基于Python实现椭圆拟合

(对应的其实是)

基于Python实现椭圆拟合
)]

这个粉色应该是在拟合瓶子的左下边缘。

资源下载地址:https://download.csdn.net/download/sheziqiong/85836974
资源下载地址:https://download.csdn.net/download/sheziqiong/85836974

Original: https://blog.csdn.net/sheziqiong/article/details/125542106
Author: biyezuopin
Title: 基于Python实现椭圆拟合

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/638225/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球