如何使用Keras在二值圖像上使用卷積神經網絡？

Question

我正在嘗試使用keras訓練ocr的CNN模型。 我對圖像進行了預處理，方法是將其轉換為灰度，去除噪聲然后將其轉換為二進制，因為二進制圖像在ocr中效果更好。 但是我得到的問題是二進制圖像具有2維，沒有通道維，而keras中的conv2d（一般來說，任何conv層）都需要3維。 那么，我應該怎么做才能添加尺寸但保持圖像二進制？ 我正在使用cv2進行圖像處理，因此請最好告訴使用該解決方案。 還告訴我是否正確，使用二進制圖像數據集更適合ocr。

Answer 1

我找到了解決方案。 我使用numpy函數numpy.expand_dims（）添加空尺寸。 因此它變成（width，height，1）。 這是我做的：

img = np.expand_dims(img,axis=2)