Opencvのcanny法で抽出した輪郭をカウントしたい

Question

前提・実現したいこと

opencvのcanny法でエッジを検出して、その数をカウントしようと試みています。
参考サイトのコードを参考にコードを書きましたが下記エラーが発生してしまいます。
このエラーメッセージは、おそらく閾値指定のフィルタリングの際の画像がグレースケールじゃない事が問題だと思うのですが、どうすればこの問題を解決してカウントできるのかわかりません。
よろしくお願いします。

発生している問題・エラーメッセージ

FindContours supports only CV_8UC1 images

該当のソースコード

canny2_color = (20, 255, 63)
canny_image_PIL_RGBA.paste(Image.new("RGB", canny_image_PIL_RGBA.size, canny2_color), mask=mask)
image2 = cv2.cvtColor(np.asarray(canny_image_PIL_RGBA), cv2.COLOR_RGBA2BGRA)
# ↑ここまでリンク先のコードと同じ(一部だけ記載)

# 閾値指定によるフィルタリング
_, threshold = cv2.threshold(image2, 240, 255, cv2.THRESH_BINARY)
# 輪郭を抽出
contours, _ = cv2.findContours(threshold, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
font = cv2.FONT_HERSHEY_DUPLEX
# 数の変数
num = 0
# 数を数える
for cnt in contours:
    approx = cv2.approxPolyDP(cnt, 0.01*cv2.arcLength(cnt, True), True)
    cv2.drawContours(image2, [approx], 0, (0), 2)
    x = approx.ravel()[0]
    y = approx.ravel()[1]

    if len(approx) >= 3:
        num += 1
        cv2.putText(image2, "No.{}".format(num), (x, y), font, 0.8, (0))

# 画像を出力
cv2.imwrite('xxx_xxx.png', image2)

Answer 1

参考サイトの処理の大部分は、エッジを蛍光色で表示するだけの処理です。
単純に閾値処理をしたいだけならcanny_imageをフィルタリングすればよいかと

Comments

[退会済みユーザー]

なるほどです。canny_imageの箇所以外をざっくり削除したら動きました。ありがとうございます。