按形状和尺寸分类

2024-09-30 22:12:50 发布

您现在位置:Python中文网/ 问答频道 /正文

如何使用机器学习来根据形状和大小对对象进行分类? 我说我画了一个小方块和一个小圆圈。两者的区别在于它们的形状和大小。所以给定一个图像,如何区分这些对象并返回结果。 在实际问题中,这些物体是太阳能光伏文件夹中的热点,它们是其中有缺陷的部分。我需要分类他们。那个I/P图像如下:

[这或多或少是一种方形热点:]

  1. https://i.stack.imgur.com/4JL7E.png

Tags: 对象https图像文件夹机器分类区分物体
1条回答
网友
1楼 · 发布于 2024-09-30 22:12:50

这个答案并没有说明机器学习或任何使用分类器的方法

The circles can be detected by the Hough Circle Transform from OpenCV cv2.HoughCircles():

Documentation for Hough Circles in OpenCV

注意:通过使用半径,可以调整要检测的圆的形状大小。老实说,我并没有真正理解点方格是什么,也许你可以在你的问题中展示一个典型的形象。在

如果图像中只有两种不同类型的对象,那么您可能甚至不需要分类器,因为这两个类已经被后续的图像处理分离(尽管这在很大程度上取决于您的输入图像)。在

import cv2
import numpy as np

img = cv2.imread('opencv_logo.png',0)
img = cv2.medianBlur(img,5)
cimg = cv2.cvtColor(img,cv2.COLOR_GRAY2BGR)

circles = cv2.HoughCircles(img,cv2.HOUGH_GRADIENT,1,20,
                            param1=50,param2=30,minRadius=0,maxRadius=0)

circles = np.uint16(np.around(circles))
for i in circles[0,:]:
    # draw the outer circle
    cv2.circle(cimg,(i[0],i[1]),i[2],(0,255,0),2)
    # draw the center of the circle
    cv2.circle(cimg,(i[0],i[1]),2,(0,0,255),3)

cv2.imshow('detected circles',cimg)
cv2.waitKey(0)
cv2.destroyAllWindows()

代码的结果如下

Result of the code

相关问题 更多 >