Python小技巧 顏色和類之間的對映

2021-08-19 20:10:44 字數 726 閱讀 2887

在做語義分割的時候,我們往往會遇到彩色的 ground truth 和 label map 之間的相互對映。這個時候最常用的方法就是用迴圈來做,這樣是最簡單的實現方法,但是效率低,迴圈過大會直接導致電腦卡死,而且**還不美觀;知道用矩陣來操作的話效率會高很多,但苦於不會實現。經過一番參考學習之後提供給大家乙個簡單高效的實現方法。

import numpy as np

def(gts):

colorize = np.zeros([4,3],'uint8')

colorize[0,:] = [ 30, 100, 30]

colorize[1,:] = [200, 200, 200]

colorize[2,:] = [ 0, 255, 0]

colorize[3,:] = [ 0, 0, 255]

ims = colorize[gts,:].reshape([gts.shape[0],gts.shape[1],3])

return ims

對於乙個有4個類的影象來說,輸入gts為一張只含有 label 的影象, 輸出ims為對映後的彩色影象,非常簡潔美觀,而且效率也高。

做了乙個測試,500x500 的 label map,對於用for迴圈來實現對映的方法,10張影象的每張平均**時間為 0.412 秒,相同條件下用上述方法時間為 0.003 秒。

python類之間關係 python 類之間的關係

類與類之間的關係 1.依賴關係 class plant def init self,zhonglei,xue,gongjili self.zhonglei zhonglei self.xue xue self.gongjili gongjili def da self,js print 植物打殭屍 ...

python管理技巧 Python小技巧整理

一 python小工具 進入相應目錄 2 字串轉換為json root mysql m echo python m json.tool job developer name 1mx male 3 批量驗證第三方庫的安裝 python c import paramiko 二 pip的高階用法 1 安裝...

Python的冷技巧小技巧

記得剛開始工作的時候,老大給我們上 c 基礎課,告訴我們字串字面量可以換行 如下 感覺真是如夢如幻。python view plain copy include int main int argc,char argv 輸出 plain view plain copy hello world.後來在寫...