在目录下新建文件,命名为predict_camera.py,完成识别。
首先导入需要的类,包括OpenCV、NumPy和TensorFlow。
import os
import cv2
import sys
import time
import numpy as np
import tensorflow as tf
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
接下来需要封装一个函数,作用是将输入的RGB图像转换为二值化图像,并将转换后的二值化图像返回。
将输入的彩色图像转换为二值化图
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