在深度學(xué)習(xí)中,Tensor的通道順序是NCHW罪塔,即channel first, 那么需要用transpose()和reshape()兩個方法來調(diào)整圖像數(shù)據(jù)格式投蝉。cv2.transpose()函數(shù)的效果
用cv2.tranpose(image,(2,0,1))無法實(shí)現(xiàn)通道順序更改
image.transpose((2,0,1))成功實(shí)現(xiàn)數(shù)據(jù)布局順序改變
用reshape()方法匹配AI模型輸入要求的維度
OpenCV的默認(rèn)顏色順序是BGR征堪,用cv2.cvtColor(image, cv2.COLOR_BGR2RGB)轉(zhuǎn)換顏色通道順序
圖像數(shù)據(jù)通道格式:NCHW和NHWC的區(qū)別