[Data] Python 이미지 여백 지우기 (numpy, mask, ...)

💡 본 문서는 'Python 이미지 여백 지우기 (numpy, mask, ...)'에 대해 정리해놓은 글입니다.
이미지 자체의 여백을 없애거나 이미지를 병합하는 과정에서 생긴 여백을 지우고 싶은 경우, 다음의 코드를 참고하여 후처리를 진행하시면 되니 참고하시기 바랍니다.

검은색 여백 제거하기

def crop_image_only_outside(img,tol=0):
    # img is 2D image data
    # tol  is tolerance
    mask = img > tol
    if img.ndim==3:
      mask = mask.all(2)
    m,n = mask.shape
    mask0,mask1 = mask.any(0),mask.any(1)
    col_start,col_end = mask0.argmax(),n-mask0[::-1].argmax()
    row_start,row_end = mask1.argmax(),m-mask1[::-1].argmax()
    return img[row_start:row_end,col_start:col_end]

흰색 여백 제거하기

def crop_image_only_outside(img,tol=255):
    # img is 2D image data
    # tol  is tolerance
    mask = img < tol
    if img.ndim==3:
      mask = mask.all(2)
    m,n = mask.shape
    mask0,mask1 = mask.any(0),mask.any(1)
    col_start,col_end = mask0.argmax(),n-mask0[::-1].argmax()
    row_start,row_end = mask1.argmax(),m-mask1[::-1].argmax()
    return img[row_start:row_end,col_start:col_end]

참고

• [StackOverFlow] Removing white space around a saved image: https://stackoverflow.com/questions/11837979/removing-white-space-around-a-saved-image
• [StackExchange] Crop black border of image using NumPy: https://codereview.stackexchange.com/questions/132914/crop-black-border-of-image-using-numpy