试了下百度PaddleHub开源的mask detection项目

Posted 2021-04-01 猪日天爱吃饭

上周看到百度AI发布的开源demo项目，用于检测是否佩戴口罩。具体的手把手教地址如下：

https://aistudio.baidu.com/aistudio/projectdetail/267322

基于python很简单的一个操作，操作思路就是载入图片===>加载预训练的模型===>输出检测结果

import paddlehub as hubimport matplotlib.pyplot as pltimport matplotlib.image as mpimgimport os


# 待预测图片root=r'./data'test_img_path = [root + os.sep + file for file in os.listdir(root)]

img = mpimg.imread(test_img_path[0])
# 展示待预测图片plt.figure(figsize=(10,10))plt.imshow(img)plt.axis('off')plt.show()

#加载预训练模型module = hub.Module(name="pyramidbox_lite_mobile_mask")# 将模型保存在test_program文件夹之中module.processor.save_inference_model(dirname=r"D:人工智障2020新冠肺炎aidu口罩检测model")

input_dict = {"image": test_img_path}
# 口罩检测预测results = module.face_detection(data=input_dict)for result in results: print(result)

# 预测结果展示img = mpimg.imread(r'./detection_result1.png')plt.figure(figsize=(10,10))plt.imshow(img)plt.axis('off')plt.show()

在网上随便找了几张图片，具体的测试结果如下所示，效果还可以：

当然，除了以大厂为代表的的开源项目外，也有其他简单的方法可以达到同样的目的。我也看到b站有up主基于opencv，利用简单的规则和图像处理手段（无于深度学习），也实现了简单动态监测个体是否佩戴口罩的目标。

贴上地址：

https://www.bilibili.com/video/av87813962from=search&seid=13106801531435502995

其实我主要是来问各位有现成的轮子吗？ ？ ？ ！ ！ ！ 如摘要所示。 。 。

在这里不抱希望地小声问一句：有针对公园场景的图片语义分割的现成轮子吗？（已经尝试了基于deeplab，cityscape数据及/ADE20K数据集的轮子，但效果不好）