grpc_client.py 725 B

1234567891011121314151617181920212223242526
  1. import grpc
  2. import sys
  3. import os
  4. sys.path.insert(0, os.path.join(os.path.dirname(__file__), "proto"))
  5. from proto import face_recognition_pb2, face_recognition_pb2_grpc
  6. # Load the test image
  7. with open("sample_aladdin.jpg", "rb") as f:
  8. image_bytes = f.read()
  9. # Connect to gRPC server
  10. channel = grpc.insecure_channel("localhost:50051")
  11. stub = face_recognition_pb2_grpc.FaceRecognitionServiceStub(channel)
  12. # Create request with model_name = "Facenet"
  13. request = face_recognition_pb2.FaceRequest(
  14. image=image_bytes,
  15. model_name="OpenFace" # specify the model here
  16. )
  17. # Call the Recognize RPC
  18. response = stub.Recognize(request)
  19. # Print results
  20. print("Name:", response.name)
  21. print("Confidence:", response.confidence)