检测人脸是否佩戴口罩需要用到的是人工智能的paddlehub模块中的pyramidbox_lite_mobile_mask模型。

代码传送门:

import paddlehub as hub
import matplotlib.pyplot as plt
import matplotlib.image as mpimg

# 待预测图图片
test_img_path = ["1.jpg"]

# 载入模型
module = hub.Module(name = "pyramidbox_lite_mobile_mask")

# 预测
input_dict = {"image": test_img_path}
results = module.face_detection(data = input_dict)

# 结果展示
img = mpimg.imread("1.jpg")
plt.figure(figsize = (10, 10))
plt.imshow(img)
plt.axis('off')
plt.show()

极简两行代码:

import paddlehub as hub

# 载入模型
module = hub.Module(name = "pyramidbox_lite_mobile_mask")

results = module.face_detection(data = {"image": ["1.jpg"]})

效果展示:

红色框代表没有佩戴,绿色框代表已经佩戴口罩,后面的百分比代表概率

从测试结果可以看到,这个模型还是准确度很高。 

Logo

开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!

更多推荐