컴퓨터 비전 알고리즘 구현 - 3. 단순 이진화

이번에는 임계치를 주었을때, 그 임계치 기준으로 영상을 이진화하는 함수를 구현하였다.

 

관련 내용은 본 블로그나 타인 내용 참고

ref : throwexception.tistory.com/607?category=873104

 

 

 

함수 구현 내용은 아래와 같다.

 

그냥 th를 넘으면 1, 아니면 0을 주는게 끝

 

def Threshold(img=None, th=125):

    if type(img) is not np.ndarray:
        raise AssertionError("img is not ndarray")
    
    row, col = img.shape
    res = np.zeros((row, col))
    for i in range(0,row):
        for j in range(0,col):
            if img[i, j] >= th:
                res[i, j] = 1
            else:
                res[i, j] = 0
    
    return res

 

 

 

 

 

 

 

맨 마지막 셀은 임계치 별 결과를 플로팅 시켰다.