【精度向上】虫の個数カウント計コード

Question

捕虫シートに捕まった虫のカウントをしようと思って以下のプログラムを書いてみました。
実際にカウントはできましたが、虫が連なりあってるところが１つにカウントされています。
画像を添付しようとはしたのですが容量的に添付できませんでした。
カメラの精度の問題もあるかと思いますが、プログラムで工夫できるところがあれば教えていただきたいです。
ご協力よろしくお願いいたします。

import cv2  #OpenCVのインポート

fname=""#開く画像ファイル名
threshold=130 #二値化閾値

img_color= cv2.imread(fname) #画像を読み出しオブジェクトimg_colorに代入
img_gray = cv2.imread(fname,cv2.IMREAD_GRAYSCALE) #画像をグレースケールで読み出しオブジェクトimg_grayに代入

ret, img_binary= cv2.threshold(img_gray, threshold, 255, cv2.THRESH_BINARY) #オブジェクトimg_grayを閾値threshold(127)で二値化しimg_binaryに代入
contours, hierarchy = cv2.findContours(img_binary, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE) #img_binaryを輪郭抽出
cv2.drawContours(img_color, contours, -1, (0,0,255), 2) #抽出した輪郭を赤色でimg_colorに重ね書き
print(len(contours)) #抽出した輪郭の個数を表示する

size=(1500,800)
re_img=cv2.resize(img_color,size)

cv2.imshow("contours",re_img) #別ウィンドウを開き(ウィンドウ名 "contours")オブジェクトimg_colorを表示
waitKey(0) #キー入力待ち
cv2.destroyAllWindows() #ウインドウを閉じる

Comments

[tiitoi]

質問欄に画像を添付してください

[yuki23]

いや、捕虫シートに絡まった虫の画像なんて添付されても困りますが（笑）

[fana]

例えば２値化結果画像とかから出発しても済む内容ならば，ダイレクトな元画像ではなくてそっちを貼るとかで比較的マシになるかも？？
（なんだろう，虫の「個数」という言葉に，とても冷たいものを感じる…）

[TakaiY]

全部の画像でなくて、問題になっている複数が連なっていて1つにカウントされてしまう部分だけを切り出した画像を貼ってみたらどうでしょう？

[ken248]

皆様ありがとうございます。
範囲を狭めて、画質落とすと伝えたいことが伝わらなくってしまうレベルの写真になってしまいます。
皆様けんとうしていただきありがとうございます。

Answer 1

「捕虫シート」と検索すると、結構なインパクトを与えられる画像が出てきます。見る限りコバエのようなものからハチのような少し大きい虫もくっついています。どのくらいをイメージして重なり想定をしているのでしょうか？

案1)　背景を消す
たぶん黄色の背景だと思います。HSVスケールに画像を変換して、色味が黄色となる背景部分を抽出して下さい。抽出した部分をマスク画像（例えば0で塗りつぶす用データ）にします。
そのまま二値化するよりは分離しやすい画像にするためです。

案2)　Scikit-imageに頼る
実際の画像がないと何とも言えませんが、こちらのサンプルのページを見るとそれっぽいものがたくさんあると思います。watershedやSLIC、distance mapなどが使えそうな気がします。

案3)　物体検出に頼る
Haar-likeを使って虫を物体検出させます。

案4)　深層学習に頼る
インスタンスセグメンテーションが使えると思います。

Comments

[ken248]

ありがとうございます。
1㎜レベルの虫が捕虫シートの淵に連なってるイメージです。
その間に5mm程度の虫もおり、それらがぴったりくっついてしまっています。
その結果3匹いるところを1匹としてしか認識できないというような状況になっております。

案１
現状背景白を試していましたが、背景は黄色もあるので試させていただきます。

案２　ありがとうございます！全く知らないサイトでしたので勉強になります。
おっしゃっていただいたのをベースに検討してみます。

それでもだめでしたら…案３を検討してみます。