python ostu代码

import cv2
from matplotlib import pyplot as plt
image = cv2.imread("1.jpg")
#转化为灰度图
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
#cv2.cvtColor(p1,p2) 是颜色空间转换函数，p1是需要转换的图片，p2是转换成何种格式。
#cv2.COLOR_BGR2RGB 将BGR格式转换成RGB格式
#cv2.COLOR_BGR2GRAY 将BGR格式转换成灰度图片
#绘制灰度图
plt.subplot(311), plt.imshow(gray, "gray")
plt.title("input image"), plt.xticks([]), plt.yticks([])
#对灰度图使用otsu算法
ret1, th1 = cv2.threshold(gray, 0, 255, cv2.THRESH_OTSU)
#cv2.threshold (源图片, 阈值, 填充色, 阈值类型)
#绘制灰度直方图
plt.subplot(312), plt.hist(gray.ravel(), 256)
#标注Ostu阈值所在直线
plt.axvline(x=ret1, color='red', label='ostu')
#legend加图例
plt.legend(loc='upper right')
plt.title("Histogram"), plt.xticks([]), plt.yticks([])
#绘制二值化图像
plt.subplot(313), plt.imshow(th1, "gray")
plt.title("output image"), plt.xticks([]), plt.yticks([])
plt.show()