OpenCV-Python中没有Mat.convertTo函数怎么办?

点击上方“小白学视觉”，选择加"星标"或“置顶”

重磅干货，第一时间送达

微信公众号：OpenCV学堂

函数说明

OpenCV C++的图像对象Mat有一个函数convertTo可以把图像数据在不同的精度类型之间来回切换比如从字节到浮点数之间来回切换。非常方便，该函数的官方说明如下：

其中参数alpha可以让数据放缩到指定的范围内，比如从字节到浮点数类型

CV_8U 转换为CV_32Falpha=1.0/255.0时表示从0～255切换到0～1之间

浮点数类型到字节类型

CV_32F 转换为CV_8Ualpha=255时表示从0～1切换到0～255之间

同时该函数还有另外一个功能就是对超过范围的数据进行最大最小极限取值，低于0的取值0，大于255取值255。

但是到了Python中之后，很多人就会认为convertTo函数只是一个数据精度类型转换函数，不会太过关心细节，然后直接使用它。这样的结果往往导致一些细数据差异跟输出不一致问题出现。

C++/Python对比

对一张输入图像完成Sobel滤波操作，C++实现的代码如下：

Mat image = imread("D:/images/dannis1.png", IMREAD_GRAYSCALE);
imshow("input", image);
Mat m1, m2;
image.convertTo(m1, CV_32F, 1.0 / 255.0);
Mat gradx;
Sobel(m1, gradx, -1, 1, 0);
gradx.convertTo(m2, CV_8U, 255);
imshow("sobel", m2);

运行结果如下：

Python对应的代码如下：

import cv2 as cv
import numpy as np

image = cv.imread("D:/images/dannis1.png", cv.IMREAD_GRAYSCALE)



    
cv.imshow("input", image)
m1 = np.float32(image) / np.float(255)
gradx = cv.Sobel(m1, -1, 1, 0);
m2 = np.uint8(gradx * 255)
cv.imshow("sobel", m2)
cv.waitKey(0)
cv.destroyAllWindows()

运行结果如下：

以往的经验处理方式是先对gradx做归一化然后再显示

这个时候只需添加一行代码即可获得正确结果，把代码：

m2 = np.uint8(gradx * 255)

替换为：

cv.normalize(gradx, gradx, 0, 1.0, cv.NORM_MINMAX)m2 = np.uint8(gradx * 255)

然后再次运行，截图如下：

但是实际结果与C++并不一致，这个时候正确修改方式如下：

m2 = np.uint8(gradx * 255)

替换为：

m2 = np.uint8(np.clip(gradx * 255, 0, 255))

然后再次运行，截图如下：

Python版本结果跟C++保持一致了！这个很多书上跟博文并没有人提到，所以我写出来分享一下，这个技术细节点！

下载1：OpenCV-Contrib扩展模块中文版教程

在「小白学视觉」公众号后台回复：扩展模块中文教程，即可下载全网第一份OpenCV扩展模块教程中文版，涵盖扩展模块安装、SFM算法、立体视觉、目标跟踪、生物视觉、超分辨率处理等二十多章内容。

下载2：Python视觉实战项目52讲

在「小白学视觉」公众号后台回复：Python视觉实战项目，即可下载包括图像分割、口罩检测、车道线检测、车辆计数、添加眼线、车牌识别、字符识别、情绪检测、文本内容提取、面部识别等31个视觉实战项目，助力快速学校计算机视觉。

下载3：OpenCV实战项目20讲

在「小白学视觉」公众号后台回复：OpenCV实战项目20讲，即可下载含有20个基于OpenCV实现20个实战项目，实现OpenCV学习进阶。

交流群

欢迎加入公众号读者群一起和同行交流，目前有SLAM、三维视觉、传感器、自动驾驶、计算摄影、检测、分割、识别、医学影像、GAN、算法竞赛等微信群（以后会逐渐细分），请扫描下面微信号加群，备注：”昵称+学校/公司+研究方向“，例如：”张三 + 上海交大 + 视觉SLAM“。请按照格式备注，否则不予通过。添加成功后会根据研究方向邀请进入相关微信群。请勿在群内发送广告，否则会请出群，谢谢理解~