图像处理&&二值化

python二值化处理图片

1. 什么是二值化

图像二值化就是将图像上的像素点的“灰度值”设置为[0, 0, 0]或[255, 255, 255]，即要么纯黑，要么纯白。

2. 二值化的作用

通过二值化，能更好地分析物体的形状和轮廓。

3. 二值化的实现

二值化的实现一般有： 全局阈值法、自适应阈值法、OTSU二值化等

4.本文将采用OTSU进行二值化处理

import cv2
img = cv2.imread("E:/data4/train/get_origin/label_188.png")
print(img)
#先进行灰度化处理，再进行二值化
Grayimg = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
#127是二值化阈值，大于255的像素值都置为0
ret, thresh = cv2.threshold(Grayimg, 0, 255, cv2.THRESH_BINARY + cv2.THRESH_OTSU)
#新建一个文件夹并用imwrite的方法对图片命名
cv2.imwrite('E:/data4/train/mask1/188.png', thresh)

原图
必须要将掩码文件需要转成黑白的，就是二值化，不然后面训练测试模型是全黑的！！！

二值化的图片如下（此种方法仅适用于处理单个图片）

批量处理（处理文件夹下的多张图片）

import os
import cv2
def binarization():    
    # 获取目录下所有图片名    
    filename = os.listdir(r"E:\Apple\test\test_label")
    print(filename)
    # os.listdir() 方法用于返回指定的文件夹包含的文件或文件夹的名字的列表。
    base_dir = r"E:\Apple\test\test_label" # input
    new_dir  = r"E:\Apple\test\test_label1" #output
    for img in filename:
      name = img
      path1= os.path.join(base_dir,img)
      img = cv2.imread(path1)
      # print(img)
      Grayimg = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
      ret, thresh = cv2.threshold(Grayimg, 0, 255,cv2.THRESH_BINARY + cv2.THRESH_OTSU)
      cv2.imwrite('img.png', thresh)
      image = Image.open('img.png')
      # 有需要可对图像进行大小调整
      image = image.resize((256, 256),Image.ANTIALIAS)
      path= os.path.join(new_dir,name)
      image.save(path)
 
binarization()

参考：
https://blog.csdn.net/m0_61899108/article/details/123006032
https://blog.csdn.net/qq_41915623/article/details/124548002