我正在尝试使用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)
答案 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')