numpy 三維矩陣下采樣小技巧

問題描述 我們有一個(160, 192, 224)的三維矩陣,由於其過大,跑神經網絡時顯存不夠。此時我們的一個思路就是對其進行一個簡單的下采樣,即沒3x3x3的小方格里只取其中間的那個數。現在的問題是如何快速高效地實現這個計算,而不是使用多個for循環。 解決方案 down_sample = rawdata[::3,::3,::3] # 這裏的rawdata是(160, 192, 224)的矩陣。
相關文章
相關標籤/搜索