import cv2
import numpy as np
#from cv2.dnn import dnn
def get_face_data():
net = cv2.dnn.readNetFromTensorflow('modelkm.pb')
font = cv2.FONT_HERSHEY_SIMPLEX
count = 1
#image = cv2.imread('.//imagesop//validation//k//image01.jpeg')
image = cv2.imread('.//imagesop//validation//nk//image3n.jpeg')
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
image = cv2.resize(image, (150, 150))
img_tensor = cv2.dnn.blobFromImage(image, 1 / 255.0, (0, 0), swapRB=False, crop=False)
net.setInput(img_tensor)
ln = net.getUnconnectedOutLayersNames()
prediction = net.forward(ln)
# print(prediction[0][0])
if prediction[0][0]>0.5:
result='no'
else:
result='yes'
print(result)
# cv2.imshow("aaa",image)
# cv2.waitKey(1000)
# cv2.destroyAllWindows()
if __name__=='__main__':
get_face_data()
米藻琴海记已识别
最新推荐文章于 2024-09-13 16:12:12 发布