opencv: 阈值处理(cv2.threshold) 探究(图示+源码)

# coding=utf-8

origin_pic = './pic/1.jpg'
save_folder = './generated_pics'

# 新建文件夹，用于存放新生成的阈值图片
import shutil
try:
    shutil.rmtree(save_folder)
except OSError:
    pass
import os
os.makedirs(save_folder)

# 读取待处理图片，并转换成单通道图片
import cv2
img = cv2.imread(origin_pic)
imgray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# 对 cv2.threshold 的 type 参数进行 0~6 的实验
import numpy as np
for type in range(0, 5, 1):
    _, thresh = cv2.threshold(src=imgray, thresh=150, maxval=100, type=type)
    concat_pic = np.concatenate([imgray, thresh], axis=1)
    cv2.imwrite('./generated_pics/type={}.jpg'.format(type), concat_pic)