python - 寻找物体中心: showing wrong coordinate outside of the target object

Question

我按照this link中的代码进行操作找到我的灰度图像的对象中心，

def find_center(im):
    immat = im
    (X, Y) = [im.shape[0],im.shape[1]]
    m = np.zeros((X, Y))

    for x in range(X):
        for y in range(Y):
            m[x, y] = immat[(x, y)] != 0
    m = m / np.sum(np.sum(m))


    # marginal distributions
    dx = np.sum(m, 1)
    dy = np.sum(m, 0)

    # expected values
    cx = np.sum(dx * np.arange(X))
    cy = np.sum(dy * np.arange(Y))
    return [cx,cy]

xy1=find_center(img)  #img is a binary image, object has value==1 and back ground value of 0
print xy1
plt.imshow(img)
plt.annotate('center', xy1, xycoords='data',
             xytext=(0.5, 0.5), textcoords='figure fraction',
             arrowprops=dict(arrowstyle="->"))
plt.show()

但是，我没有得到正确的答案(中心不在对象内部)，下图显示了输出:

我做错了什么？

Answer 1

我认为函数find_center正在正确计算中心坐标。不过，有一种更优雅、更有效的方法来执行此计算(请参阅下面的代码片段)。

问题是您将 xy1，即 [cx, cy] 传递给 plt.annotate，但您需要传递 [cy，cx]。如果您将代码更改为 xy1 = find_center(img)[::-1]，问题应该会得到解决。

尝试一下这段代码:

import numpy as np
from skimage import io
import matplotlib.pyplot as plt
from matplotlib.patches import Circle

triskele = io.imread('/image/fczjh.png')
img = triskele > 0

[cx, cy] = np.transpose(np.nonzero(img)).mean(axis=0)

fig, ax = plt.subplots(1)
ax.imshow(img, cmap=plt.cm.gray)
ax.axis('off')
ax.add_patch(Circle((cy, cx), radius=12, color='red'))
plt.show(fig)