大家好,这里是小白三岁,三岁做梦都没有想到,能够踏上AI技术路,用上AI还完成了一个理想丰满现实骨感的项目,项目漏洞极多,希望大佬们帮忙改进,谢谢大家~~
项目简介
三岁通过AI技术对导入的照片进行AI识别,查看是否戴口罩,如果没有就采用人脸关键点检测,在对应的位置上加上“口罩”,里面涉及了百度飞桨的两款AI识别接口(点击查看具体的内容)
项目不足
因为小编三岁,技术不行对图片处理包括.jpg和.png格式的图处理不好,导致了照片底黑,过大,还希望有大佬帮忙改进提供自己的版本!在这里谢谢大家啦,如果有其他的想法可以自行修改添加自己的版本。
点击查看项目
项目说明
项目思路
- 项目先对导入照片文件进行读取,获得照片的绝对路径
- 建立连接根据照片路径 对路径下的图片进行口罩识别
- 对于返回内容(未带口罩的:‘NO MASK’)进行下一步
- 对未带口罩的进行关键点识别
- 获得相应的关键点
- 根据关键点添加“口罩”
- 继续进行判定直到全部判定结束
代码详解
- 有关库的导入及使用(该处以飞桨AI studio平台代码使用方法为准,如有问题欢迎私聊)
下载有关模块:
#运行模块加载
#口罩识别
!hub install pyramidbox_lite_mobile_mask==1.1.0
#关键点检测
!hub install face_landmark_localization==1.0.2
#抠图
!hub isntall deeplabv3p_xception65_humanseg
#关键点识别
!hub install face_landmark_localization==1.0.2
# hub run face_landmark_localization --input_path "/PATH/TO/IMAGE"
#下载第三方包
!pip install matplotlib
- 导入有关第三方库:
import paddlehub as hub
import cv2
import os
import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import paddlehub as hub
import numpy as np
import math
from PIL import Image
import importlib
- 创建照片地址读取函数:
def photo_path_read(path, add)