1. import matplotlib.pyplot as plt 
2. import matplotlib.image as mpimg 
3. 
4. image_str='test1011'   # test010  test101
5. test_img_path = ["./%s.jpg"%(image_str)]
6. img = mpimg.imread(test_img_path[0]) 
7. 
8. # 展示待预测图片
9. plt.imshow(img) 
10. plt.axis('off') 
11. plt.show()
12. 
13. 
14. 
15. import matplotlib.pyplot as plt
16. import matplotlib.image as mpimg
17. import paddlehub as hub
18. module = hub.Module(name="ultra_light_fast_generic_face_detector_1mb_640")
19. 
20. # 待预测图片
21. test_img_path = ["./crowd1.jpg"]
22. img = mpimg.imread(test_img_path[0])
23. 
24. # 展示待预测图片
25. plt.imshow(img)
26. plt.axis('off')
27. plt.title('Face mask detection based on artificial intelligence algorithm by Jason Niu')
28. plt.show()
29. 
30. 
31. input_dict = {"image": test_img_path}
32. # execute predict and print the result
33. results = module.face_detection(data=input_dict)
34. 
35. for result in results:
36. print(result)
37. 
38. # 预测结果展示
39. img = mpimg.imread("./face_detector_640_predict_output/crowd1.jpg")
40. plt.imshow(img)
41. plt.axis('off')
42. plt.title('Face mask detection based on artificial intelligence algorithm by Jason Niu')
43. plt.show()