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