问题描述
可视化过滤器的时候,遇到了图片不能正常显示的问题
本该出现有颜色和纹理过滤器图片,但是出现了大量白色图片,有些图片上还有零零散散的点,出错位置代码如下。
layer_name = 'block3_conv1'
size = 64
margin = 5
result = np.zeros((8 * size + 7 * margin, 8 * size + 7 * margin, 3))
for i in range(8):
for j in range(8):
filter_img = generate_pattern(layer_name, i+j*8, size=size)
horizontal_start = i * size + i * margin
horizontal_end = horizontal_start + size
vertical_start = j * size + j * margin
vertical_end = vertical_start + size
result[horizontal_start : horizontal_end,
vertical_start : vertical_end, :] = filter_img
plt.figure(figsize=(20,20))
plt.imshow(result)
plt.show()
原因分析:
代码的目的是将一张张图片(filter_img)拼接到同一张图片(result)上一起显示,一切看着都是那么正确,但就是显示不出来正确的图片。
其实很简单,问题就出在下面这段赋值语句
result[horizontal_start : horizontal_end,
vertical_start : vertical_end, :] = filter_img
读者的result数组元素可能是浮点数,而filter_img数组的元素是整数,这样的话,赋值语句过后,result数组中虽然赋值成功了,但是全部都是浮点数,这样的图片是显示不出来的。读者自己想办法修改,或者参考我的修改方式:
result = np.zeros((8 * size + 7 * margin, 8 * size + 7 * margin, 3)).astype('uint8') # 如果不加uint8,会出现整数数组赋值给浮点数数组的情况,此时赋值成功但是浮点数组不能显示为图像