PaddleHub实现人脸检测

最新推荐文章于 2024-08-09 09:28:32 发布

little_baby1029

最新推荐文章于 2024-08-09 09:28:32 发布

阅读量471

点赞数 1

文章标签：深度学习

本文链接：https://blog.csdn.net/xingzhe1993/article/details/107643904

版权

!pip install paddlehub==1.6.2 -i https://pypi.tuna.tsinghua.edu.cn/simple
import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import paddlehub as hub

test_img_path = ["timg.jpg"]

img = mpimg.imread(test_img_path[0])

# 展示待预测图片
# plt.figure(figsize=(10, 10))
# plt.imshow(img)
# plt.axis('off')
# plt.show()

# 读取待预测图片文件
with open('test.txt', 'r') as f:
    test_img_path=[]
    for line in f:
        test_img_path.append(line.strip())
print(test_img_path)

# 加载模型
#在预测时会将图片输入缩放为640 * 480
module = hub.Module(name="ultra_light_fast_generic_face_detector_1mb_640")
input_dict = {"image": test_img_path}

results = module.face_detection(data=input_dict, visualization=True)
for result in results:
    print(result)

# 预测结果展示
for img_name in test_img_path:
    img = mpimg.imread("face_detector_640_predict_output/" + img_name)
    plt.figure(figsize=(10,10))
    plt.imshow(img)
    plt.axis('off')
    plt.show()