检测人脸是否佩戴口罩需要用到的是人工智能的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"]})
效果展示:
红色框代表没有佩戴,绿色框代表已经佩戴口罩,后面的百分比代表概率
从测试结果可以看到,这个模型还是准确度很高。