Object Detection을 진행한 이미지에서 bndbox 영역만 crop 한 후classfication 진행.

원본 이미지를 학습된 classifier에 넣고 Grad-Cam 적용. 

Cam 이미지에서 다시 bndbox 영역을 Crop 한 후 원본 이미지에 Paste 해주기.

 

import os
from PIL import Image

# Get annotation
bndbox_area = (xmin, ymin, xmax, ymax)

# image crop
crop_image = cam_image.crop(bndbox_area)

# paste image
origin_image.paste(crop_image, (xmin, ymin))

# save image
origin_image.save('./result/{0}'.format(cam_images))

 

+ Recent posts