opencvで色付きの外接矩形を表示したい

Question

前提・実現したいこと

python+opencvで開発を行っています。
画像中の物体の輪郭を抽出して、抽出した輪郭に対して外接矩形を入れたいのですが、
その外接矩形に赤色をつけたいです。
現状、黒色の外接矩形しか表示されません。

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

輪郭を抽出後、外接矩形を表示したところグレースケールで表示されてしまう。

該当のソースコード

import cv2

def main():
    src = cv2.imread('C:/Users/ito/Anaconda3/envs/Sample/pic/IMG_0050.jpg')
    
    height, width, channels = src.shape
    image_size = height * width
    
    src_gray = cv2.cvtColor(src, cv2.COLOR_RGB2GRAY)
    
    ret, dst = cv2.threshold(src_gray, 122, 255, cv2.THRESH_BINARY)
    
    dst, contours, hierarchy = cv2.findContours(dst, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
    
    for i, contour in enumerate(contours):
        area = cv2.contourArea(contour)
        if area < 500:
            continue
        
        if image_size * 0.99 < area:
            continue
        
        x,y,w,h = cv2.boundingRect(contour)
        rec = cv2.rectangle(dst, (x,y),(x+w,y+h),(0,0,255), 2)
    
    cv2.imwrite('C:/Users/ito/Anaconda3/envs/Sample/pic/sampplepgm.png', rec)
    
if __name__ == '__main__':
    main()

補足情報（FW/ツールのバージョンなど）

windows7 professional
python3.6.5
opencv3.3.1

Answer 1

dstがgrayスケールの画像なので、そこに3チャネルのカラー情報を載せるのは無理です。dstを3チャネルデータに戻すか、代わりにsrcを使ってください。

dst3ch = cv2.cvtColor(dst, cv2.COLOR_GRAY2RGB)
rec = cv2.rectangle(dst3ch, (x,y),(x+w,y+h),(0,0,255), 2)

or

rec = cv2.rectangle(src, (x,y),(x+w,y+h),(0,0,255), 2)

Comments

[Taka11]

ご回答ありがとうございます！

Answer 2

グレースケールの画像行列に色付きの書き込みをすることはできません。
forループに突入する前に、再度画像行列をRGBに直してください。

dst = cv2.cvtColor(dst, cv2.COLOR_GRAY2RGB)
for i, contour in enumerate(contours):
    ...

    rec = cv2.rectangle(dst, (x,y),(x+w,y+h),(0,0,255), 2)

もちろん元々の画像の色は復元されませんのでご安心を。

---

レナ姐さんで試したらこんな感じになりました。

Comments

[Taka11]

ご回答ありがとうございます！
試したところ無事できました。
ありがとうございます！