ND4J讀取圖片

一、構建加載器

NativaImageLoader loader = new NativeImageLoader(height, width, channels);

構建圖片加載器，順便完成了對圖片的縮放功能，可以直接用於fit網絡模型。

1. height: 圖片的高
2. width: 圖片的寬
3. channels: 圖片通道數，3爲彩色，1爲黑白

二、圖片讀取

INDArray image = loader.asMatrix(new File("/picture/path"))

根據圖片路徑讀取對應圖片成爲INDArray

三、圖片矩陣規範化

DataNormalization scaler = new ImagePreProcessScaler(0, 1);
scaler.transform(image);

ImagePreProcessScaler構造器參數爲需要規範化的區間。

---

更多文檔可以查看 https://github.com/sjsdfg/deeplearning4j-issues。
你的star是我持續分享的動力