numpy 的陣列內容比對

max-stackoverflow — Fri, 02 Jul 2021 18:07:10 +0000

最近看到別人分享的一個副程式，因為想要修改，可是「完全」無法理解，為什麼可以這樣去使用，原來是變成了自定義的物件，所以可以取出特定項目進行一一的比對。

先來看看原本的程式碼：

import numpy as np

def changeColor(im, from_rgb_value, to_rgb_value):
    data = np.array(im)
    red, green, blue = data[:,:,0], data[:,:,1], data[:,:,2]
    r1, g1, b1 = from_rgb_value
    mask = (red == r1) & (green == g1) & (blue == b1)

看到這裡，沒想到 red == r1 的比對是 array 裡的項目一一比對，而不是2個boolean object 內容的比對。而比對的結果居然還可以放回去 mask array 裡，這個真的很神奇。

PIL 的 getpixel 似乎略慢

max-stackoverflow — Tue, 20 Oct 2020 17:30:53 +0000

由於想針對影像做處理，使用 loop 去拿image 裡的pixel 使用 image.getpixel 似乎太沒有效率，於是使用 numpy.asarray() 把 PIL image object 轉成 array.

使用的範例圖片 width: 921, height 1001, 服用下面的 code 後：

import PIL
import numpy
from PIL import Image
im = Image.open('U_51234.bmp')
data = numpy.asarray(im)
print("shape:", data.shape)
print("100,150:",data[100,150])
print("100,150:",data[100][150])

取得 shape: (1001, 921)

所以是先 h 再 w, 如果要直接 access array 需要先放 y axis ，再使用 x axis, 與原先的 getpixel(x,y) 順序會相反。

Convert between PIL image and NumPy ndarray

image = Image.open(“max-demo.jpg”)   # image is a PIL image 
array = numpy.array(image)          # array is a numpy array 
image2 = Image.fromarray(array)   # image2 is a PIL image

Convert between PIL image and PyOpenCV matrix

image = Image.open(“max-demo.jpg”)                  # image is a PIL image
mat = pyopencv.Mat.from_pil_image(image)  # mat is a PyOpenCV matrix 
image2 = mat.to_pil_image()                        # image2 is a PIL image

Convert between OpenCV image and NumPy ndarray

cimg = cv.LoadImage("max-demo.jpg", cv.CV_LOAD_IMAGE_COLOR)   # cimg is a OpenCV image 
pimg = Image.fromstring("RGB", cv.GetSize(cimg), cimg.tostring())  # pimg is a PIL image 
array = numpy.array(pimg)     # array is a numpy array 
pimg2 = cv.fromarray(array)    # pimg2 is a OpenCV image

numpy – Max的程式語言筆記

numpy 的陣列內容比對

PIL 的 getpixel 似乎略慢