从图像文件中读取并处理
在mnist_predict目录下新建文件,命名为read_pic.py,使用OpenCV读取新图像,并进行预处理,在PyCharm中编写如下代码。
import os
import cv2
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
将输入的彩色图像转换为二值化图
def color_input(endimg):
# 灰度化转换
img_gray = cv2.cvtColor(endimg, cv2.COLOR_BGR2GRAY)
ret, img_threshold = cv2.threshold(img_gray, 127, 255, cv2.THRESH_BINARY_INV)
return img_threshold
读取图像并显示
def read_pic(path):
img = cv2.imread(path, cv2.IMREAD_COLOR)
cv2.imshow('img', img)
cv2.waitKey(0)
img_threshold = color_input(img)
cv2.imshow('img_threshold', img_threshold)
cv2.waitKey(0)
if name == 'main':
read_pic("pic.png")
运行以上代码,在read_pic()函数中可读取与read_pic.py同级目录下的pic.png图像文件并且显示出来。pic.png图像文件如图8-8所示,这是手写的一些数字,关掉显示框之后,调用color_input()函数,并将读取的图像传递进去。