Преобразование изображения PIL в GTK Pixbuf

Я смотрю, есть ли другой способ конвертировать изображение PIL в GTK Pixbuf. Сейчас все, что у меня есть, - это то, что похоже на неэффективную практику кодирования, которую я нашел и взломал для моих нужд. Это то, что у меня есть до сих пор:

def image2pixbuf(self,im):  
    file1 = StringIO.StringIO()  
    im.save(file1, "ppm")  
    contents = file1.getvalue()  
    file1.close()  
    loader = gtk.gdk.PixbufLoader("pnm")  
    loader.write(contents, len(contents))  
    pixbuf = loader.get_pixbuf()  
    loader.close()  
    return pixbuf 

Есть ли более простой способ сделать это преобразование, которое я пропустил?

Ответ 1

Вы можете сделать это эффективно, если вы пройдете через массив numpy:

import numpy
arr = numpy.array(im)
return gtk.gdk.pixbuf_new_from_array(arr, gtk.gdk.COLORSPACE_RGB, 8)

Ответ 2

Если вы используете PyGI и GTK + 3, здесь альтернатива, которая также устраняет необходимость в зависимости от numpy:

import array
from gi.repository import GdkPixbuf

def image2pixbuf(self,im):
    arr = array.array('B', im.tostring())
    width, height = im.size
    return GdkPixbuf.Pixbuf.new_from_data(arr, GdkPixbuf.Colorspace.RGB,
                                          True, 8, width, height, width * 4)

Ответ 3

Я не могу использовать gtk 3.14 (эта версия имеет метод new_from_bytes) [1], так что этот workaroud как ваш, чтобы заставить его работать:

from gi.repository import GdkPixbuf
import cv2

def image2pixbuf(im): 
  # convert image from BRG to RGB (pnm uses RGB)
  im2 = cv2.cvtColor(im, cv2.COLOR_BGR2RGB)
  # get image dimensions (depth is not used)
  height, width, depth = im2.shape
  pixl = GdkPixbuf.PixbufLoader.new_with_type('pnm')
  # P6 is the magic number of PNM format, 
  # and 255 is the max color allowed, see [2]
  pixl.write("P6 %d %d 255 " % (width, height) + im2.tostring())
  pix = pixl.get_pixbuf()
  pixl.close()
  return pix

Литература: