comment convertir une image RVB en tableau numpy?

Question

J'ai une image RVB. Je veux le convertir en tableau numpy. j'ai fait ce qui suit

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

Il crée un tableau sans forme. Je suppose que c'est un objet iplimage.

Answer 1

Vous pouvez utiliser une nouvelle interface python OpenCV (si je ne me trompe pas, elle est disponible depuis OpenCV 2.2). Il utilise nativement des tableaux numpy :

 import cv2
im = cv2.imread("abc.tiff",mode='RGB')
print type(im)

résultat:

 <type 'numpy.ndarray'>

Answer 2

PIL (Python Imaging Library) et Numpy fonctionnent bien ensemble.

J'utilise les fonctions suivantes.

 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 )

Le 'Image.fromarray' est un peu moche parce que je coupe les données entrantes à [0,255], les convertit en octets, puis crée une image en niveaux de gris. Je travaille principalement en gris.

Une image RVB serait quelque chose comme :

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

Answer 3

Vous pouvez également utiliser matplotlib pour cela.

 from matplotlib.image import imread

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

production: <class 'numpy.ndarray'>

Answer 4

À partir d'aujourd'hui, votre meilleur pari est d'utiliser :

 img = cv2.imread(image_path)   # reads an image in the BGR format
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)   # BGR -> RGB

Vous verrez que img sera un tableau numpy de type :

 <class 'numpy.ndarray'>

Answer 5

Réponse tardive, mais j'en suis venu à préférer le imageio aux autres alternatives

 import imageio
im = imageio.imread('abc.tiff')

Semblable à cv2.imread() , il produit un tableau numpy par défaut, mais sous forme RVB.