opencv_tutorials/python/code_042/opencv_042.py at master · HZHCoder1990/opencv_tutorials · GitHub

32 lines (27 loc) · 878 Bytes

import cv2 as cv

import numpy as np

# THRESH_BINARY = 0

# THRESH_BINARY_INV = 1

# THRESH_TRUNC = 2

# THRESH_TOZERO = 3

# THRESH_TOZERO_INV = 4

src = cv.imread("lena.png")

cv.namedWindow("input", cv.WINDOW_AUTOSIZE)

cv.imshow("input", src)

h, w = src.shape[:2]

# 自动阈值分割 OTSU

gray = cv.cvtColor(src, cv.COLOR_BGR2GRAY)

ret, binary = cv.threshold(gray, 0, 255, cv.THRESH_BINARY | cv.THRESH_OTSU)

print("ret :", ret)

cv.imshow("binary", binary)

result = np.zeros([h, w*2, 3], dtype=src.dtype)

result[0:h,0:w,:] = src

result[0:h,w:2*w,:] = cv.cvtColor(binary, cv.COLOR_GRAY2BGR)

cv.putText(result, "input", (10, 30), cv.FONT_ITALIC, 1.0, (0, 0, 255), 2)

cv.putText(result, "binary, threshold = " + str(ret), (w+10, 30), cv.FONT_ITALIC, 1.0, (0, 0, 255), 2)

cv.imshow("result", result)

cv.imwrite("binary_result.png", result)

cv.waitKey(0)

cv.destroyAllWindows()

Expand file tree