[DIP] opencv : Region of Interest (ROI) : cv2.selectROI

cv2에서 사각형의 ROI를 선택하는 가장 쉬운 방법.

 

Signature

ret_val = cv2.selectROI(
    [window_name], 
    img 
    [, showCrossHair=True, fromCenter=False]
    )

• window_name : ROI 선택을 수행할 window이름. str
• img : 보여질 이미지.
• showCrossHair : ROI 중심에 십자모양 표시 여부
• fromCenter : 마우스 시작지점을 영역의 중심으로 간주
• ret_val = (x,y,w,h) of ROI

Sample code

import cv2
import numpy as np
import os

d_path = os.path.dirname(__file__)
img_path = f'{d_path}/lena.png'

img = cv2.imread(img_path)
cv2.imshow('img', img)
x,y,w,h = cv2.selectROI('img', img, False)

print(cv2.getWindowProperty('img', cv2.WND_PROP_VISIBLE))
if w and h: #선택시 0이 아닌 수로 w,h가 설정됨.
    roi = img[y:y+h, x:x+w]
    cv2.imshow('roi', roi)
    cv2.moveWindow('roi',0,0)
    cv2.imwrite(f'{d_path}/roi2.png', roi)
    print('Enter ESC to quit!')

while cv2.getWindowProperty('img', cv2.WND_PROP_VISIBLE) >=1: # check whether x button is clicked.
    # c = input('Quit? (enter q)')
    # if c == 'q':
    #     break
    key_code = cv2.waitKey(50)&0xff #50msec
    if key_code == 27: #check whether ESC key is entered.
        break


cv2.destroyAllWindows()

• 선택시 space or Enter 를 누르면 됨.
• 선택을 취소하고 싶을 경우 c키 누름 : 이 경우, ret_val는 모두 0이 됨.

아래 그림에서 푸른색의 사각형이 선택한 ROI임.