come convertire un'immagine RGB in array numpy?

Question

Ho un'immagine RGB. Voglio convertirlo in array numpy. Ho fatto il seguente

im = cv.LoadImage("abc.tiff") 
a = numpy.asarray(im) 

crea un array senza forma. Suppongo sia un oggetto iplimage.

Come fare?

Grazie

Answer 1

È possibile utilizzare più recente interfaccia OpenCV python (se non mi sbaglio è disponibile dal OpenCV 2.2). Esso utilizza nativamente array NumPy:

import cv2 
im = cv2.imread("abc.tiff") 
print type(im) 

risultato:

<type 'numpy.ndarray'> 

Answer 2

È necessario utilizzare cv.LoadImageM invece di cv.LoadImage:

In [1]: import cv 
In [2]: import numpy as np 
In [3]: x = cv.LoadImageM('im.tif') 
In [4]: im = np.asarray(x) 
In [5]: im.shape 
Out[5]: (487, 650, 3) 

Answer 3

def opencv_image_as_array(im): 
    """Interface image from OpenCV's native format to a numpy array. 

    note: this is a slicing trick, and modifying the output array will also change 
    the OpenCV image data. if you want a copy, use .copy() method on the array! 
    """ 
    import numpy as np 
    w, h, n = im.width, im.height, im.channels 
    modes = {1:"L", 3:"RGB"}#, 4:"RGBA"} 
    if n not in modes: 
    raise StandardError('unsupported number of channels: {0}'.format(n)) 
    out = np.asarray(im) if n == 1 else np.asarray(im)[:,:,::-1] ## BGR -> RGB 
    return out 

Answer 4

PIL (Python Imaging Library) e Numpy funzionano bene insieme.

Uso le seguenti funzioni.

from PIL import Image 
import numpy as np 

def load_image(infilename) : 
    img = Image.open(infilename) 
    img.load() 
    data = np.asarray(img, dtype="int32") 
    return data 

def save_image(npdata, outfilename) : 
    img = Image.fromarray(np.asarray(np.clip(npdata,0,255), dtype="uint8"), "L") 
    img.save(outfilename) 

Il 'Image.fromarray' è un po 'brutto perché ho clip dati in arrivo a [0255], convertire in byte, quindi creare un'immagine in scala di grigi. Per lo più lavoro in grigio.

immagine An RGB sarebbe qualcosa di simile:

outimg = Image.fromarray(ycc_uint8, "RGB") 
outimg.save("ycc.tif") 

Answer 5

Quando si utilizza la risposta di David Poole ho uno SystemError con scala di grigi PNG e forse altri file. La mia soluzione è:

import numpy as np 
from PIL import Image 

img = Image.open(filename) 
try: 
    data = np.asarray(img, dtype='uint8') 
except SystemError: 
    data = np.asarray(img.getdata(), dtype='uint8') 

realtà img.getdata() avrebbe funzionato per tutti i file, ma è più lento, così ho usarlo solo quando l'altro metodo non riesce.

Answer 6

Per questo è anche possibile utilizzare matplotlib.

import matplotlib.image as mpimg 

img = mpimg.imread('abc.tiff') 
print(type(img)) 

uscita: <class 'numpy.ndarray'>