複数のマスク画像に対してforループなしで重心を求めたい

Question

複数の2値化したモノクロマスクをN枚まとめた3次元Numpy配列[x, y, N]に対してOpencvの重心を求めるcv2.moments関数を各マスクに適用したいと思います。
forループを使えば各画像を取り出して適用することは難しく無いですが、numpyの性質を使ったエレガントな手法はありませんでしょうか？

Comments

[tiitoi]

OpenCV の関数 cv2.moments がそのような入力に対応してないので、for で回すしかありません。

Answer 1

def get_centroids(imgs):
    """
    Args:
        imgs (np.ndarray): in shape (N, H, W)
    Returns:
        centroids (np.ndarray): [y, x] order, in shape (N, 2)
    """
    N = imgs.shape[0]
    X = np.array(zip(*np.where(imgs > 0))).rehshape(N, -1, 2)
    centeroids = X.mean(axis=1).astype(np.int32)
    return centroids

Answer 2

使ったことはないですが、numpy.apply_along_axis はどうでしょう？
参考