按形状和大小分类

Question

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

[这或多或少是一个正方形的热点：]

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

Answer 1

此答案并未详细说明机器学习或任何使用分类器的方法

圆圈可以被Hough Circle Transform检测到 来自 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()

代码结果如下