【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的結果圖。