CV:基于人工智能算法實(shí)現(xiàn)人臉口罩的實(shí)時(shí)檢測(cè)(結(jié)合無(wú)人機(jī)可,實(shí)現(xiàn)實(shí)時(shí)警告提醒)
輸出結(jié)果
實(shí)現(xiàn)代碼
參考文章:CV:基于人工智能算法實(shí)現(xiàn)人臉口罩的實(shí)時(shí)檢測(cè)(結(jié)合無(wú)人機(jī)可,實(shí)現(xiàn)實(shí)時(shí)警告提醒)——代碼
更新中……
import matplotlib.pyplot as plt
import matplotlib.image as mpimg
image_str='test1011' # test010 test101
test_img_path = ["./%s.jpg"%(image_str)]
img = mpimg.imread(test_img_path[0])
# 展示待預(yù)測(cè)圖片
plt.imshow(img)
plt.axis('off')
plt.show()
import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import paddlehub as hub
module = hub.Module(name="ultra_light_fast_generic_face_detector_1mb_640")
# 待預(yù)測(cè)圖片
test_img_path = ["./crowd1.jpg"]
img = mpimg.imread(test_img_path[0])
# 展示待預(yù)測(cè)圖片
plt.imshow(img)
plt.axis('off')
plt.title('Face mask detection based on artificial intelligence algorithm by Jason Niu')
plt.show()
input_dict = {"image": test_img_path}
# execute predict and print the result
results = module.face_detection(data=input_dict)
for result in results:
print(result)
# 預(yù)測(cè)結(jié)果展示
img = mpimg.imread("./face_detector_640_predict_output/crowd1.jpg")
plt.imshow(img)
plt.axis('off')
plt.title('Face mask detection based on artificial intelligence algorithm by Jason Niu')
plt.show()