【PaddlePaddle】百度口罩检测Win10平台效果测试

一、安装PaddlePaddle

请参考链接：【PaddlePaddle】Win10 Anaconda安装PaddlePaddle。

二、安装PaddleHub

pip install --upgrade paddlehub -i https://pypi.tuna.tsinghua.edu.cn/simple

三、测试

PaddleHub口罩检测提供了两种预训练模型，pyramidbox_lite_mobile_mask和pyramidbox_lite_server_mask。

3.1 单张图片测试

import paddlehub as hub

test_img_path = ["./test_mask_detection.jpg"]
module = hub.Module(name='pyramidbox_lite_server_mask', version='1.1.0')
input_dict = {'image': test_img_path}
results = module.face_detection(data=input_dict, use_multi_scale=True, shrink=0.6)
for result in results:
    print(result)

3.2 批量测试

将所有待测图片的路径放在一个txt文件。

import paddlehub as hub

with open('test.txt', 'r') as f:
    test_img_path = []
    for line in f:
        test_img_path.append(line.strip())
#print(test_img_path)

module = hub.Module(name='pyramidbox_lite_server_mask', version='1.1.0')
input_dict = {'image': test_img_path}
results = module.face_detection(data=input_dict, use_multi_scale=True, shrink=0.6)
for result in results:
    print(result)

PaddleHub会自动下载预训练模型，存放在C:\Users\xxx\.paddlehub\modules下。

运行成功会打印出boundingbox的信息。

并且会在当前路径下新建一个detection_result文件夹，里面是画了boundingbox的结果图。