OpenCV在一个圆圈上获取像素

时间:2018-11-23 10:12:02

标签: python-3.x opencv3.0

我是OpenCV的新手,我正在尝试从图像中获取圆的像素。

例如,我在随机图像上画一个圆:

import cv2

raw_img = cv2.imread('sample_picture.png')
x = 50
y = 50
rad = 20
cv2.circle(raw_img,(x,y),rad,(0,255,0),-1)
cv2.imshow('output', raw_img)
cv2.waitKey(0)
cv2.destroyAllWindows()

输出显示带圆圈的图像。 但是,我希望能够以数组的形式返回圆上的所有像素。有什么办法吗?我知道我可以从圆公式中获得近似坐标,但是它将涉及很多十进制计算,而且我很确定函数 cv2.circle() >已经计算了像素,是否有办法从函数本身中获取像素,而不是计算自己的像素?

此外,如果有可能,我想使用 cv2.ellipse() 作为一个坐标数组来获取椭圆的像素。但是这次,我只想从椭圆的一部分(从某个角度到另一个角度,可以在cv2.ellipse()的参数中指定)获取像素。

谢谢。

1 个答案:

答案 0 :(得分:1)

您可以使用numpy函数实现所需的功能:

numpy.where(condition[, x, y])

链接中的功能的详细说明:https://docs.scipy.org/doc/numpy-1.15.1/reference/generated/numpy.where.html

在您的情况下,您希望它返回具有非零值的坐标。使用此方法,您可以在空数组上绘制任何内容,它将返回与非零对应的所有行和列。

它将返回满足您设置的条件的数组的索引。以下是显示用法示例的代码。

import cv2
import numpy as np

raw_img = cv2.imread('sample_picture.png')
x = 50
y = 50
rad = 20
cv2.circle(raw_img,(x,y),rad,(0,255,0),-1)

# Here is where you can obtain the coordinate you are looking for
combined = raw_img[:,:,0] + raw_img[:,:,1] + raw_img[:,:,2]
rows, cols, channel = np.where(combined > 0)    

cv2.imshow('output', raw_img)
cv2.waitKey(0)
cv2.destroyAllWindows()