File size: 2,130 Bytes
c5380c5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 |
from fastapi import FastAPI
from pydantic import BaseModel, HttpUrl
import requests
import face_recognition
import pickle
import cv2
import pyrebase
import os
# Initialize Firebase using Pyrebase
config = {
"apiKey": "AIzaSyClnRJAnrJgAgkYjuYnlvu-CJ6Cxyklebo",
"databaseURL": "https://console.firebase.google.com/project/socioverse-2025/database/socioverse-2025-default-rtdb/data/~2F",
"authDomain": "socioverse-2025.firebaseapp.com",
"projectId": "socioverse-2025",
"storageBucket": "socioverse-2025.appspot.com",
"messagingSenderId": "689574504641",
"appId": "1:689574504641:web:a22f6a2fa343e4221acc40",
"serviceAccount":"socioverse-2025-firebase-adminsdk-gcc6m-6bfb53e6d9.json"
}
firebase = pyrebase.initialize_app(config)
storage = firebase.storage()
storage.child().download("Faces/pkl/face_encodings.pkl","face_encodings.pkl")
app = FastAPI()
class ImgInput(BaseModel):
image_url: HttpUrl
class ImgOutput(BaseModel):
label: str
def recognize_face(image_url: HttpUrl) -> ImgOutput:
# Downloading image
response = requests.get(image_url)
with open("examp.jpg", 'wb') as file:
file.write(response.content)
# Load the stored face encodings and labels from the pickle file
with open("face_encodings.pkl", "rb") as file:
data = pickle.load(file)
face_encodings = data["encodings"]
labels = data["labels"]
# Load a new image you want to recognize
new_image = cv2.imread("examp.jpg")
new_face_encoding = face_recognition.face_encodings(new_image)
if len(new_face_encoding) == 0:
print("No faces found in the new image.")
else:
# Compare the new face encoding to the stored encodings
results = face_recognition.compare_faces(face_encodings, new_face_encoding[0])
os.remove("examp.jpg")
for i, result in enumerate(results):
if result:
return ImgOutput(label=labels[i])
return ImgOutput(label="unable to detect")
@app.post('/')
async def scoring_endpoint(item:ImgInput):
result = recognize_face(item.image_url)
return result |