我需要检测形状并计算图像中每个形状的出现。我最初检测到轮廓并对它们进行近似,并计算出每个轮廓中的顶点。我的代码如下所示:
import cv2
import numpy as np
import collections
import sys
img = cv2.imread(str(sys.argv[1]),0)
ret,thresh = cv2.threshold(img,127,255,0)
contours,hierarchy = cv2.findContours(thresh,1,2)
no_of_vertices = []
i = 0
mask = np.zeros(img.shape,np.uint8)
for contour in contours:
cnt = contour
area = cv2.contourArea(cnt)
if area>150:
epsilon = 0.02*cv2.arcLength(cnt,True)
approx = cv2.approxPolyDP(cnt,epsilon,True)
no_of_vertices.append(len(approx))
counter = collections.Counter(no_of_vertices)
a,b = counter.keys(),counter.values()
i=0
while i<len(counter):
print a[i],b[i]
i = i + 1
我的代码无法检测此图片中的星星:
我应该在代码中做出哪些更改?
答案 0 :(得分:3)
对我来说有用的是对形状周边区域的平方根进行比较。它对于一颗恒星来说约为0.145(+/- .0015,因为有些边缘并没有完美地出现)。六边形为0.255,三角形为.21,正方形为.247,五边形为.250。
圆度也起作用(三角形在0.26到.27之间),它的区别相似(六边形为.83,三角形为.55-.56,正方形为.77,和.78)五角大楼)
下面是它的C ++代码(我的PC上没有python,但这个想法是相同的):
#include "stdafx.h"
#include <opencv/cxcore.h>
#include <opencv2\core\mat.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <iostream>
#include <opencv/cxcore.h>
#include <opencv/highgui.h>
#include <opencv/cv.h>
#include <opencv2/opencv.hpp>
#include <opencv2/core/core.hpp>
using namespace cv;
using namespace std;
RNG rngee(12345);
int main() {
Mat im = imread("C:/this/is.a/path/image.png", CV_LOAD_IMAGE_COLOR);
Mat imgrey = im.clone();
cvtColor(im, imgrey, CV_RGB2GRAY);
vector<vector<Point> > imContours;
vector<Vec4i> hierarchy;
double divMaxSize = 0.175, divMinSize = 0.125;
namedWindow("Image", CV_WINDOW_NORMAL| CV_WINDOW_KEEPRATIO | CV_GUI_EXPANDED);
threshold(imgrey, imgrey, 100, 255, 0);
findContours(imgrey, imContours, hierarchy, CV_RETR_TREE, CV_CHAIN_APPROX_SIMPLE, Point(0, 0) );
for (int i=0; i < imContours.size(); i++) {
Scalar color = Scalar( rngee.uniform(0, 255), rngee.uniform(0,255), rngee.uniform(0,255) );
cout << "sqrt(Area)/arcLength = " << sqrt(contourArea(imContours[i]))/arcLength(imContours[i], true ) << endl;
if(sqrt(contourArea(imContours[i]))/arcLength(imContours[i], true ) < divMaxSize && sqrt(contourArea(imContours[i]))/arcLength( imContours[i], true ) > divMinSize)
{
drawContours(im, imContours, i, color, 2, 8, hierarchy, 0, Point() );
cout << "I'm a star!" << endl;
}
imshow("Image", im);
waitKey(0);
}
imshow("Image", im);
waitKey(0);
}
两种方式 - 使用圆形或我的sqrt(区域)/ arclength方法 - 导致: