numpy.vectorize(),詳情見文檔
實現(xiàn):
img = np.array(Image.open(ori_path))
img_cat = np.vectorize(lab.id2trainId.get)(img)
這里我是把矩陣的值進行一對一映射,其中l(wèi)ab.id2trainId是一個字典,dictionary的get方法可以返回指定鍵的值,這樣就構(gòu)成了映射關(guān)系,處理速度很快。
除了一對一單值映射之外,比如把一個值映射為一個tuple也是可以的,這個tuple可以是RGB值,這樣就可以把原本一個矩陣以帶顏色的圖片形式顯示了。