[bug] numpy.astype(uint8)和opencv imwrite函数的自动数据转换

- 日理万妓 2022-07-26 05:18 12阅读 0赞

**现象描述**：

我在实现图像去模糊功能的时候，利用维纳滤波方法，因为对图像进行数字处理之后，其中数据格式变成了float32，其中有一些像素的值小于0或大于255，这个时候我利用numpy中的astype(uint8)进行类型转换，然后保存图片，结果如下图的右侧子图所示，出现了原则上不应该出现的红色边缘，这主要是由于红色通道图像出现了不再\[0，255\]区间的像素值，而这个时候我如果用astype(uint8)进行处理的话，数据就会出现截断或溢出，具体见图2实验。

![Center][] ![Center 1][]

![Image 1][]

直接保存结果 先转换数据astype()

图1

![Image 1][]

![Center 2][]

图2

OpenCV 中imwrite函数对数据类型进行自动转换时，是根据如下的方法进行的：

###  Saturation Arithmetics[¶][Link 1] ###

As a computer vision library, OpenCV deals a lot with image pixels that are often encoded in a compact, 8- or 16-bit per channel, form and thus have a limited value range. Furthermore, certain operations on images, like color space conversions, brightness/contrast adjustments, sharpening, complex interpolation (bi-cubic, Lanczos) can produce values out of the available range. If you just store the lowest 8 (16) bits of the result, this results in visual artifacts and may affect a further image analysis. To solve this problem, the so-called *saturation* arithmetics is used. For example, to store  r, the result of an operation, to an 8-bit image, you find the nearest value within the 0..255 range:

![I(x,y)= \\min ( \\max (\\textrm\{round\}(r), 0), 255)][Image 1]

Similar rules are applied to 8-bit signed, 16-bit signed and unsigned types. This semantics is used everywhere in the library. In C++ code, it is done using the  saturate\_cast<> functions that resemble standard C++ cast operations. See below the implementation of the formula provided above:

I.at<uchar>(y, x) = saturate_cast<uchar>(r);

where cv::uchar  is an OpenCV 8-bit unsigned integer type. In the optimized SIMD code, such SSE2 instructions as paddusb, packuswb, and so on are used. They help achieve exactly the same behavior as in C++ code.

Note:

Saturation is not applied when the result is 32-bit integer.

[Center]: https://img-blog.csdn.net/20160407212229657?watermark/2/text/aHR0cDovL2Jsb2cuY3Nkbi5uZXQv/font/5a6L5L2T/fontsize/400/fill/I0JBQkFCMA==/dissolve/70/gravity/Center
[Center 1]: /images/20220724/ef30f9637c0c44dbbb6cca97e52c7ab0.png
[Image 1]: 
[Center 2]: /images/20220724/8a2b7c71a4e941bcbcb24aa308f3c29c.png
[Link 1]: http://docs.opencv.org/2.4/modules/core/doc/intro.html#saturation-arithmetics