【人工智能】python代码检测人脸是否佩戴口罩,paddlehub
检测人脸是否佩戴口罩需要用到的是人工智能的paddlehub模块中的pyramidbox_lite_mobile_mask模型。代码传送门:import paddlehub as hubimport matplotlib.pyplot as pltimport matplotlib.image as mpimg# 待预测图图片test_img_path = ["1.jpg"]...
·
检测人脸是否佩戴口罩需要用到的是人工智能的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"]})
效果展示:
红色框代表没有佩戴,绿色框代表已经佩戴口罩,后面的百分比代表概率
从测试结果可以看到,这个模型还是准确度很高。
开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!
更多推荐
已为社区贡献3条内容
所有评论(0)