图像识别口罩

我要开发同款
OX2021年04月13日
267阅读

作品详情

基于CNN卷机神经网络的图像识别,识别口罩采用tensorflow简单的识别数据集(200幅戴口罩图片,200幅没戴口罩图片)搭建网络 训练模型 因为深度网络训练 200张数据集远远不够 但是可以进行简单的测试搭建网络 训练模型我做这个项目时 因为是测试 只进行了五次迭代选择从测试集测试代码:import cv2from tensorflow.keras.models import load_modelresize = 224label = ('戴口罩!', '没戴口罩')image = cv2.resize(cv2.imread('./test/test1.jpg'),(resize,resize))image = image.astype("float") / 255.0image = image.reshape((1, image.shape[0], image.shape[1], image.shape[2]))model = load_model('./model/mask.h5')predict = model.predict(image)i = predict.argmax(axis=1)[0]print(label[i])
查看全文
声明:本文仅代表作者观点,不代表本站立场。如果侵犯到您的合法权益,请联系我们删除侵权资源!如果遇到资源链接失效,请您通过评论或工单的方式通知管理员。未经允许,不得转载,本站所有资源文章禁止商业使用运营!
下载安装【程序员客栈】APP
实时对接需求、及时收发消息、丰富的开放项目需求、随时随地查看项目状态

评论