detection한 이미지 Crop하는 방법

보통 객체 탐지 모델의 결과는 bbox의 네 꼭지점을 주는 경우가 많은데요. 

네 꼭지점에 맞게 이미지를 한번 잘라보겠습니다. 

 

• 필요한것 : 객체 탐지 모델, 이미지

저는 인물 탐지하는 faster-rcnn 모델을 사용했습니다. 

먼저 이미지를 모델에 거쳐 결과를 받아주세요.

 

저는 장원영 사진으로 하겠습니다ㅎㅅㅎ

출처:장원영인스타

[6.4021919e+01, 3.3994653e+02, 6.4194012e+02, 1.1920336e+03, 9.9788922e-01]

위의 결과는 각각 [left, up, right, down, prob]입니다.

여기서는 bbox만 필요하기 때문에 맨 뒤 prob는 빼고 사용해주면 됩니다. 

from PIL import Image

image1 = Image.open("test.png") #이미지를 불러옴

croppedImage=image1.crop((left,up,right,down)) #잘라줌

정말 간단하죠?

왼쪽은 원본, 오른쪽은 crop된 이미지입니다.