numpy矢量化后opencv显示黑色图像

时间:2013-01-08 21:04:52

标签: python opencv numpy

我正在尝试使用np的矢量化但是imshow正在显示一个黑色图像,如果我正确理解矢量化它应该是白色的。我认为问题是输出类型,但我不能让它工作。

import numpy as np
import cv2
class Test():
    def run(self):        
        arr = np.zeros((25,25))
        arr[:]=255
        cv2.imshow('white',arr)
        flatarr = np.reshape(arr,25*25)
        vfunc = np.vectorize(self.func)
        #vfunc = np.vectorize(self.func,otypes=[np.int])#same effect
        flatres = vfunc(flatarr)
        shouldbewhite = np.reshape(flatres,(25,25))
        cv2.imshow('shouldbewhite',shouldbewhite)        
    def func(self,a):
        return 255
cv2.namedWindow('white',0)
cv2.namedWindow('shouldbewhite',0)
a = Test()
a.run()
cv2.waitKey(0)

1 个答案:

答案 0 :(得分:4)

来自docs

  

函数imshow在指定的窗口中显示图像。如果   窗口是用CV_WINDOW_AUTOSIZE标志创建的,图像是   显示其原始大小。否则,缩放图像以适合   窗户。该功能可能会缩放图像,具体取决于其深度:

     
      
  • 如果图像是8位无符号,则按原样显示。
  •   
  • 如果图像是16位无符号或32位整数,则像素除以256.即,值范围[0,255 * 256]映射到[0,255]。
  •   
  • 如果图像是32位浮点,则像素值乘以255   是,值范围[0,1]映射到[0,255]。
  •   

如果您运行以下代码:

class Test():
    def run(self):        
        arr = np.zeros((25,25))
        arr[:]=255
        print arr.dtype
        flatarr = np.reshape(arr,25*25)
        vfunc = np.vectorize(self.func)
        flatres = vfunc(flatarr)
        print flatres.dtype
        shouldbewhite = np.reshape(flatres,(25,25))
        print shouldbewhite.dtype
    def func(self,a):
        return 255

你会得到类似的东西:

float64
int32
int32

所以你的第二种情况除以256,它是整数除法,它四舍五入为0.尝试用

vfunc = np.vectorize(self.func,otypes=[np.uint8])

您可能还想考虑用

替换第一个数组
arr = np.zeros((25,25), dtype='uint8')