AI计算机视觉进阶项目(一)——带口罩识别检测(2)

官方合作微信:gldz_super

本专栏《AI计算机视觉进阶项目》主要以计算机视觉实战项目为主,第一个项目为口罩检测:该项目将分为几个模块进行展示

1.项目概述

        要求如下:检测出视频中的人是否带口罩,如果带了进行检测,该项目主要分为正常佩戴、未佩戴口罩以及不规范(漏鼻子)三个类别。训练的数据集有正常佩戴1915张、未佩戴口罩1918张以及不规范(漏鼻子)1919张数据。

2.本节任务

  1. 对所有的图像首先将人脸裁剪出来,然后进行Blob转换(上节是对单个图像进行处理)
  2. 将数据存储为numpy格式

3.项目实现

3.1导入需要的库

import cv2
import numpy as np
import matplotlib.pyplot as plt
import os, glob
import tqdm

3.2加载路径中的所有类别图像并加载模型

 # 处理所有图像
    path = 'images/'
    # 剪切人脸,为了图像的计算效率对人的图像进行剪切
    # 加载SSD模型:将人脸剪切出来
    face_detector = cv2.dnn.readNetFromCaffe('./weights/deploy.prototxt.txt',
                                             'weights/res10_3

你可能感兴趣的:(AI计算机视觉项目实战进阶,人工智能,口罩检测)