从scikit阵列转换为PIL photoimage的图像失真

时间:2019-04-28 15:16:07

标签: python-3.x tkinter python-imaging-library scikit-image

我正在尝试将由scikit-image和scipy处理的图像添加到tkinter gui中。要将其添加到画布,需要将其保存为png或转换为PIL图像。但是,当我尝试使用ImageTk的{​​{1}}时,图像会失真很多。我不希望将其另存为png,因为这只是生成数据标签的中间步骤。

我尝试检查数组的形状,它们是相同的。我尝试打印出图像,并且Image.fromarray()失真时,filled_objects是正确的图像。因此,在Tkinter gui中这不是问题。另外,如果我不使用im,它将产生相同的输出。

np.asarray()

这是两张图片,def generateCanny(imageName): #imagename should be a path to the image, created with os path join img = skimage.io.imread(imageName) print('orig {}'.format(img.shape)) gray = np.sqrt((img*img).sum(-1)) #converts the image to greyscale edges = skimage.feature.canny(gray, sigma=3) fill = scipy.ndimage.binary_fill_holes(edges) return fill imageName = os.path.join(imagePath, imageStr) filled_objects = generateCanny(imageName) a = np.asarray(filled_objects) im = PIL.Image.fromarray(a) 在左侧,im在右侧

scikit to PIL image conversion distortion

我认为您可以轻松地进行转换,因为filled_objects只是一个数组,但是filled_objects必须进行一些处理。

2 个答案:

答案 0 :(得分:1)

问题是fromarray不能正确解释布尔数组a。如果您使用以下方法将a转换回RGB,

# Extend the array into 3 dimensions, repeating the data:
a = np.repeat(a[...,None],3,axis=2).astype(np.uint8)
# Scale to 0-255:
a = 255*a
im = PIL.Image.fromarray(a)

然后im.show()将显示正确的图像。

答案 1 :(得分:0)

将结果转换为NumPy的uint8将达到目的:

from skimage import data, color, feature, util
import tkinter as tk
import numpy as np
from PIL import ImageTk, Image
from scipy.ndimage import binary_fill_holes

rgb = data.hubble_deep_field()
gray = color.rgb2grey(rgb)
edges = feature.canny(gray, sigma=3)
filled_objects = binary_fill_holes(edges)

img_bool = Image.fromarray(filled_objects)
img_uint8 = Image.fromarray(util.img_as_ubyte(filled_objects))

root = tk.Tk()
photo_bool = ImageTk.PhotoImage(img_bool)
photo_uint8 = ImageTk.PhotoImage(img_uint8)
label_bool = tk.Label(root, image=photo_bool).grid(row=1, column=1)
label_uint8 = tk.Label(root, image=photo_uint8).grid(row=1, column=2)
root.mainloop()

result