OpenCV安装及使用

安装命令：

 pip install opencv-python    -i https://pypi.douban.com/simple --trusted-host=pypi.douban.com

使用CV2调整图片大小，作为模型的输入

import cv2
import tensorflow as tf
import numpy as np

img_data = cv2.imread("/path/to/digit.jpeg", cv2.IMREAD_GRAYSCALE)
print(img_data.shape)
#ouput: (651, 650)
# cv2.imshow("img title",img5)
# cv2.waitKey(0)
img_data = cv2.resize(img_data, (28, 28))


model = tf.keras.models.load_model("/path/to/model.h5")
prob = model.predict(img_data.reshape((-1, 28, 28)))
print(np.argmax(prob))