[v0.8]영상처리_OpenCV_이미지 내 관심영역 (ROI)

211230 작성

<본 블로그는 귀퉁이 서재님의 블로그를 참고해서 공부하며 작성하였습니다>

https://bkshin.tistory.com/entry/OpenCV-6-dd?category=1148027 

OpenCV - 6. 이미지 내 관심영역(Region of Interest, ROI) 표시

 

 

 

1. 관심 영억 (ROI, Region Of Interest) 표시하기

: 영상 내에서 관심이 있는 영역

 

cv2.rectangle(roi, (0,0), (h-1, w-1), (0,255,0))

: 태양 부분만 슬라이싱 한 roi 이미지의 (0, 0)부터 (h-1, w-1)까지 (0, 255, 0) 색으로 사각형을 표시

: (0, 0)은 좌측 상단

: (h-1, w-1)은 우측 하단

=> roi 이미지의 좌측 상단부터 우측 하단까지를 꼭짓점으로 갖는 사각형을 그려준다

: RGB 값인 (0, 255, 0)은 녹색

# 관심영역 표시

import cv2
import numpy as np

img = cv2.imread('img/sun.jpg')

x=320; y=150; w=50; h=50        # roi 좌표
roi = img[y:y+h, x:x+w]         # roi 지정        ---①

print(roi.shape)                # roi shape, (50,50,3)
cv2.rectangle(roi, (0,0), (h-1, w-1), (0,255,0), 10) # roi 전체에 사각형 그리기 ---②
cv2.imshow("img", img)

key = cv2.waitKey(0)
print(key)
cv2.destroyAllWindows()

해 ROI 영역에 넣기

 

 

+) 특정 영역 확대 및 복제

# 관심영역 복제 및 새 창에 띄우기

import cv2
import numpy as np

img = cv2.imread('img/sun.jpg')

x=320; y=150; w=50; h=50
roi = img[y:y+h, x:x+w]     # roi 지정
img2 = roi.copy()           # roi 배열 복제 ---①

img[y:y+h, x+w:x+w+w] = roi # 새로운 좌표에 roi 추가, 태양 2개 만들기
cv2.rectangle(img, (x,y), (x+w+w, y+h), (0,255,0), 10) # 2개의 태양 영역에 사각형 표시

cv2.imshow("img", img)      # 원본 이미지 출력
cv2.imshow("roi", img2)     # roi 만 따로 출력

cv2.waitKey(0)
cv2.destroyAllWindows()

특정 영역 강조 및 새 창 열기

 

 

2. 마우스 드래그로 관심 영역 표시

onMouse()라는 콜백 함수

: (1) 마우스 왼쪽 버튼을 눌렀을 때, (2) 누른 상태로 드래그했을 때, (3) 마우스 왼쪽 버튼을 뗄 때

(x0, y0)

: 마우스 드래그가 시작된 위치

(x, y)

: 마우스의 현재 위치

# 마우스로 관심영역 지정 및 표시, 저장 

import cv2
import numpy as np

isDragging = False                      # 마우스 드래그 상태 저장 
x0, y0, w, h = -1,-1,-1,-1              # 영역 선택 좌표 저장
blue, red = (255,0,0),(0,0,255)         # 색상 값 

def onMouse(event,x,y,flags,param):     # 마우스 이벤트 핸들 함수  ---①
    global isDragging, x0, y0, img      # 전역변수 참조
    if event == cv2.EVENT_LBUTTONDOWN:  # 왼쪽 마우스 버튼 다운, 드래그 시작 ---②
        isDragging = True
        x0 = x
        y0 = y
    elif event == cv2.EVENT_MOUSEMOVE:  # 마우스 움직임 ---③
        if isDragging:                  # 드래그 진행 중
            img_draw = img.copy()       # 사각형 그림 표현을 위한 이미지 복제
            cv2.rectangle(img_draw, (x0, y0), (x, y), blue, 2) # 드래그 진행 영역 표시
            cv2.imshow('img', img_draw) # 사각형 표시된 그림 화면 출력
    elif event == cv2.EVENT_LBUTTONUP:  # 왼쪽 마우스 버튼 업 ---④
        if isDragging:                  # 드래그 중지
            isDragging = False          
            w = x - x0                  # 드래그 영역 폭 계산
            h = y - y0                  # 드래그 영역 높이 계산
            print("x:%d, y:%d, w:%d, h:%d" % (x0, y0, w, h))
            if w > 0 and h > 0:         # 폭과 높이가 양수이면 드래그 방향이 옳음 ---⑤
                img_draw = img.copy()   # 선택 영역에 사각형 그림을 표시할 이미지 복제
                # 선택 영역에 빨간 사각형 표시
                cv2.rectangle(img_draw, (x0, y0), (x, y), red, 2) 
                cv2.imshow('img', img_draw) # 빨간 사각형 그려진 이미지 화면 출력
                roi = img[y0:y0+h, x0:x0+w] # 원본 이미지에서 선택 영영만 ROI로 지정 ---⑥
                cv2.imshow('cropped', roi)  # ROI 지정 영역을 새창으로 표시
                cv2.moveWindow('cropped', 0, 0) # 새창을 화면 좌측 상단에 이동
                cv2.imwrite('img/cropped.jpg', roi)   # ROI 영역만 파일로 저장 ---⑦
                print("croped.")
            else:
                cv2.imshow('img', img)  # 드래그 방향이 잘못된 경우 사각형 그림ㅇㅣ 없는 원본 이미지 출력
                print("좌측 상단에서 우측 하단으로 영역을 드래그 하세요.")

img = cv2.imread('img/sun.jpg')
cv2.imshow('img', img)
cv2.setMouseCallback('img', onMouse) # 마우스 이벤트 등록 ---⑧
cv2.waitKey()
cv2.destroyAllWindows()

직접 마우스로 드래그!

잘린 이미지 저장

 

 

 

+) opencv의 함수 사용!

ret = cv2.selectROI(win_name, img, showCrossHair=True, fromCenter=False)

: 마우스로 관심 영역을 드래그 한 뒤 스페이스나 엔터를 누르면 새 창에 관심 영역이 뜬다
win_name : 관심영역을 표시할 창의 이름
img : 관심영역을 표시할 이미지
showCrossHair : 선택 영역 중심에 십자 모양 표시 여부
fromCenter : 마우스 시작 지점을 영역의 중심으로 지정
ret : 선택한 영역의 좌표와 크기 (x, y, w, h); 선택을 취소하면 모두 0으로 지정됨

# selectROI로 관심영역 지정 및 표시, 저장 (roi_select_img.py)

import cv2, numpy as np

img = cv2.imread('img/sun.jpg')

x,y,w,h	= cv2.selectROI('img', img, False)      # 원본 이미지 띄어줌
if w and h:
    roi = img[y:y+h, x:x+w]
    cv2.imshow('cropped', roi)  # ROI 지정 영역을 새창으로 표시
    cv2.moveWindow('cropped', 0, 0) # 새창을 화면 좌측 상단에 이동
    cv2.imwrite('img/cropped2.jpg', roi)   # ROI 영역만 파일로 저장

cv2.waitKey(0)
cv2.destroyAllWindows()

Select a ROI and then press SPACE or ENTER button!
Cancel the selection process by pressing c button!

 

opencv 함수 이용해서 마우스 드래그!

 

잘린 이미지 저장

 

 

후.. ROI

직접 마우스로 드래그 할 수 있었다니

굉장히 유용하군