pythonのエラーがわかりません。画像処理をしています。

Question

画像から一定の大きさ以下のオブジェクトを消したいと思っているのですが、エラーが出ています。
エラーの意味を教えてください。

#-*- coding:utf-8 -*-
import cv2
import numpy as np

def remove_objects(img, lower_size=None, upper_size=None):
    # find all objects
    nlabels, labels, stats, centroids = cv2.connectedComponentsWithStats(img)

    sizes = stats[1:, -1]
    _img = np.zeros((labels.shape))

    # process all objects, label=0 is background, objects are started from 1
    for i in range(1, nlabels):

        # remove small objects
        if (lower_size is not None) and (upper_size is not None):
            if lower_size < sizes[i - 1] and sizes[i - 1] < upper_size:
                _img[labels == i] = 255

        elif (lower_size is not None) and (upper_size is None):
            if lower_size < sizes[i - 1]:
                _img[labels == i] = 255

        elif (lower_size is None) and (upper_size is not None):
            if sizes[i - 1] < upper_size:
                _img[labels == i] = 255

    return _img



#if __name__ == '__main__':
img_a = cv2.imread('21/image21_4.jpg')
'''
    # change color space
    img_Lab = cv2.cvtColor(img, cv2.COLOR_BGR2Lab)

    # smoothing
    img_Lab = cv2.GaussianBlur(img_Lab, (5, 5), 3)
    img_L, img_a, img_b = cv2.split(img_Lab)
'''
    # binary transformation
    #thre, img_greenarea = cv2.threshold(img_a, 120, 255, cv2.THRESH_BINARY_INV)

    # remove very-large and very-small objects
img_greenarea_clean = remove_objects(img_a, lower_size=300, upper_size=None)

    # save
   # cv2.imwrite('output.beans_field.cleanobj.before.jpg', img_greenarea)
cv2.imwrite('21/clean.jpg', img_greenarea_clean)

エラー

File "yobi.py", line 46, in <module>
    img_greenarea_clean = remove_objects(img_a, lower_size=300, upper_size=None)
  File "yobi.py", line 7, in remove_objects
    nlabels, labels, stats, centroids = cv2.connectedComponentsWithStats(img)
cv2.error: OpenCV(4.1.0) C:\projects\opencv-python\opencv\modules\imgproc\src\connectedcomponents.cpp:3927: error: (-215:Assertion failed) L.channels() == 1 && I.channels() == 1 in function 'cv::connectedComponents_sub1'

Answer 1

connectedComponentsWithStatsの引数は 8bit1ch の画像でなければならないところに，
マルチチャンネルな（カラーの）画像を渡しているのではないでしょうか．

Comments

[fana]

CVのエラーは，メッセージを見ましょう．今回は
> channels() == 1
とか出ているので，画像のch数が問題だと推測できる．
（「channels() == 1」は「1chでなきゃならないのにそうじゃないやつがいるからダメ」という意味．）
→connectedComponentsWithStatsをリファレンスで見れば，"the 8-bit single-channel image to be labeled" と書いてあるので裏が取れる．
→自分が渡している img は何chになってるのかを確認

…という順で調べればよいです．

[na_na]

解説ありがとうございます！
グレイスケールで読み込んだところ解決いたしました。