如何批量喂食訓練有素的 Python TensorFlow model？

Question

我創建了一個自己的 model 並使用來自 Keras 的 ImageTrainGenerator- 使用flow_from_directory對其進行了訓練。

像這樣： 如何批量訓練 model 。 一切正常，我檢查了生成的批次，圖片就是它必須的樣子。

我的問題是，我想在在線人臉檢測中使用這個訓練有素的 model。 我以所需的寬度和高度裁剪面部，將其轉換為數組，但預測結果很糟糕。

我認為實時流式傳輸的圖像必須與 Imagetraingenerator 創建的（批次）相同。 知道如何將cv2.imread(path)圖像轉換成批處理來預測 class 嗎？

Answer 1

您只需添加批次尺寸即可將其轉換為具有 1 個樣本的批次： np.expand_dims(img, axis=0) 。