làm thế nào để chuyển đổi một hình ảnh RGB thành mảng numpy?

Question

Tôi có hình ảnh RGB. Tôi muốn chuyển nó thành mảng numpy. Tôi đã làm như sau

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

nó tạo ra một mảng không có hình dạng. Tôi cho rằng đó là đối tượng iplimage.

Cách thực hiện?

Cảm ơn

Answer 1

Bạn có thể sử dụng giao diện OpenCV python mới hơn (nếu tôi không nhầm nó có sẵn từ OpenCV 2.2). Nó natively sử dụng mảng NumPy:

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

kết quả:

<type 'numpy.ndarray'> 

Answer 2

Bạn cần phải sử dụng cv.LoadImageM thay vì 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) và NumPy làm việc tốt với nhau.

Tôi sử dụng các chức năng sau.

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) 

'Image.fromarray' hơi xấu vì tôi cắt dữ liệu đến [0,255], chuyển đổi thành byte, sau đó tạo hình ảnh thang độ xám. Tôi chủ yếu làm việc với màu xám.

Một RGB hình ảnh sẽ là một cái gì đó như:

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

Answer 5

Khi sử dụng câu trả lời từ David Poole tôi nhận được một SystemError với quy mô xám PNG và các file có thể khác. Giải pháp của tôi là:

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') 

Thực tế img.getdata() sẽ hoạt động cho tất cả các tệp nhưng chậm hơn, vì vậy tôi chỉ sử dụng khi phương thức khác không thành công.

Answer 6

Bạn cũng có thể sử dụng matplotlib cho việc này.

import matplotlib.image as mpimg 

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

đầu ra: <class 'numpy.ndarray'>