深度学习与OpenCV（阈值篇）

Q_ytsup5681

于 2024-06-10 14:22:04 发布

阅读量644

点赞数 16

文章标签：深度学习 opencv 人工智能

本文链接：https://blog.csdn.net/Q_ytsup5681/article/details/139576252

版权

在当今这个以数据为中心的时代，图像处理已成为许多行业中不可或缺的一环。从医学成像到自动驾驶汽车，再到安全监控系统，精确的图像分析技术是推动这些领域进步的关键因素之一。随着深度学习技术的迅猛发展，尤其是卷积神经网络（CNN）在图像识别领域的突破，我们拥有了更为强大的工具来处理和分析视觉数据。OpenCV，作为一个广泛使用的开源计算机视觉库，其与深度学习的结合，为我们提供了前所未有的机会来探索和实现复杂的图像处理功能。

一、深度学习在OpenCV中的应用

OpenCV支持多种深度学习框架，如TensorFlow和Caffe，使得在图像处理流程中集成预训练的深度学习模型变得简单。典型的应用包括：

物体检测：使用预训练的CNN模型识别图像中的多个对象。
图像分割：通过像素级别的分类来实现图像的区域分割。
风格转换：将图像内容与特定艺术风格结合，创造出新的视觉效果。
超分辨率：提高图像的分辨率，使图像更清晰。

二、作用

提高效率：自动化复杂的图像处理任务，减少手动干预。
提升准确性：与传统图像处理方法相比，深度学习模型能够学习更多的特征和模式。
扩展能力：可以轻松地对新类型的图像数据进行处理和分析。

目前比较流行的CV框架有很多，以下是其中一些：

TensorFlow：由Google开发的开源机器学习框架，其高度灵活的编程模型使得它在计算机视觉领域的应用非常广泛。TensorFlow支持多种神经网络结构和优化算法，可以快速搭建复杂的CV模型。

PyTorch：另一个由Facebook开发的开源机器学习框架，与TensorFlow相比，PyTorch更加轻量级和易于使用，因此被誉为“深度学习界的脚本语言”。PyTorch支持动态图和静态图两种模式，并且具有极其方便的数据加载和处理接口。

Keras：一种高层次的深度学习框架，可以在TensorFlow、Theano和CNTK等低级别框架上运行。Keras的主要优点是简单易用，可以快速实现常见的深度学习模型，但它缺少底层细节的控制，对于一些特定需求的研究可能不太方便。

OpenCV：一种专门针对计算机视觉应用的开源框架，它提供了大量的图像处理和计算机视觉算法和工具，如图像/视频读取、处理、显示、特征提取、物体识别等。OpenCV易于使用且功能强大，可以用于快速搭建计算机视觉应用的原型。

Torch：一种基于Lua语言的科学计算框架，主要用于机器学习和计算机视觉等领域。Torch的主要特点是容易上手和可扩展性高，但由于它使用了不太常见的Lua语言，可能不是所有开发者都适应它。

这些框架各有所长，TensorFlow和PyTorch是当前最为流行的深度学习框架，Keras则更频繁被用于快速原型搭建，OpenCV和Torch则专注于计算机视觉领域的实际应用。

以下是安装过程以及示例代码：

检查Python环境：确保你的系统中已经安装了Python，并且版本至少为3.6以上。可以通过在命令行输入python --version来检查Python的版本。
打开命令提示符：在Windows系统中，可以按下Win+R键，然后输入cmd并回车，打开命令提示符窗口。在macOS或Linux系统中，则可以直接打开终端。
执行安装命令：在命令提示符窗口中输入以下命令并回车执行
```
pip install opencv-python
```

案例：

import cv2
import numpy as np

# 加载预训练的模型
net = cv2.dnn.readNetFromCaffe("deploy.prototxt", "model.caffemodel")

# 读取图像文件
image = cv2.imread("example.jpg")

# 准备输入并进行前向传播
blob = cv2.dnn.blobFromImage(image, 1.0, (300, 300), (104.0, 177.0, 123.0))
net.setInput(blob)
detections = net.forward()

# 提取检测结果
for i in range(detections.shape[2]):
    confidence = detections[0, 0, i, 2]  # 置信度
    if confidence > 0.5:  # 设置阈值
        class_id = int(detections[0, 0, i, 1])
        # 画出检测到的物体
        box = detections[0, 0, i, 3:7] * np.array([width, height, width, height])
        (startX, startY, endX, endY) = box.astype("int")
        cv2.rectangle(image, (startX, startY), (endX, endY), (0, 255, 0), 2)

# 显示图像
cv2.imshow("Output", image)
cv2.waitKey(0)
cv2.destroyAllWindows()

阈值（Thresh）

自适应阈值

简单阈值算法使用全局阈值，但一副图像中不同位置的光照情况可能不同，全局阈值会失去很多信息。这种情况下我们需要采用自适应阈值。自适应阈值二值化函数根据图片一小块区域的值来计算对应区域的阈值，从而得到也许更为合适的图片。
dst = cv2.adaptiveThreshold(src, maxval, thresh_type, type, Block Size, C)
Block Size-它决定了计算阈值的窗口区域的大小。

C-它只是一个常数，会从平均值或加权平均值中减去该值。

下面的代码比较了具有不同照明的图像的全局阈值和自适应阈值：

	import cv2 as cv
import numpy as np
from matplotlib import pyplot as plt
img = cv.imread('sudoku.png',0)
img = cv.medianBlur(img,5)
ret,th1 = cv.threshold(img,127,255,cv.THRESH_BINARY)
th2 = cv.adaptiveThreshold(img,255,cv.ADAPTIVE_THRESH_MEAN_C,cv.THRESH_BINARY,11,2)
th3 = cv.adaptiveThreshold(img,255,cv.ADAPTIVE_THRESH_GAUSSIAN_C,cv.THRESH_BINARY,11,2)
titles = ['Original Image', 'Global Thresholding (v = 127)',
            'Adaptive Mean Thresholding', 'Adaptive Gaussian Thresholding']
images = [img, th1, th2, th3]
for i in xrange(4):
    plt.subplot(2,2,i+1),plt.imshow(images[i],'gray')
    plt.title(titles[i])
    plt.xticks([]),plt.yticks([])
plt.show()

结果：