Archived
61 lines
2.3 KiB
Python
61 lines
2.3 KiB
Python
#!/usr/bin/python3
|
|
import json
|
|
|
|
import cv2
|
|
import uvicorn, asyncio
|
|
import numpy as np
|
|
|
|
from fastapi import FastAPI, File, UploadFile, Form, HTTPException, Depends
|
|
|
|
from schemas.api_schemas import *
|
|
|
|
from workers.mongo_worker import MongoWorker
|
|
from workers.face_rec_worker import FaceRecognition
|
|
|
|
app = FastAPI()
|
|
mongo_worker = MongoWorker()
|
|
face_recognition_worker = FaceRecognition()
|
|
|
|
@app.post('/add_person')
|
|
async def add_person(json_data: str = Form(...),
|
|
img_data: UploadFile = File(...),
|
|
face_recognition: FaceRecognition = Depends(lambda: face_recognition_worker),
|
|
mongo: MongoWorker = Depends(lambda: mongo_worker)) -> BaseResponse:
|
|
json_validate = AddPersonJSON(**json.loads(json_data))
|
|
if img_data.content_type not in ALLOWED_MIME_TYPES:
|
|
raise HTTPException(400, detail="Invalid file type")
|
|
|
|
file_data = await img_data.read()
|
|
face_img = cv2.imdecode(np.fromstring(file_data, np.uint8), cv2.IMREAD_COLOR)
|
|
encodings = face_recognition.get_encodings(face_img)
|
|
if not encodings:
|
|
raise HTTPException(422, detail="Can't find human face")
|
|
|
|
result = mongo.add_person(json_validate.username, encodings[0], img_data.filename, face_img)
|
|
if result.error:
|
|
raise HTTPException(422, detail=str(result))
|
|
return BaseResponse(result="Success")
|
|
|
|
@app.post('/find_by_img')
|
|
async def find_by_img(img_data: UploadFile = File(...),
|
|
face_recognition: FaceRecognition = Depends(lambda: face_recognition_worker),
|
|
mongo: MongoWorker = Depends(lambda: mongo_worker)) -> BaseResponse:
|
|
if img_data.content_type not in ALLOWED_MIME_TYPES:
|
|
raise HTTPException(400, detail="Invalid file type")
|
|
|
|
file_data = await img_data.read()
|
|
face_img = cv2.imdecode(np.fromstring(file_data, np.uint8), cv2.IMREAD_COLOR)
|
|
encodings = face_recognition.get_encodings(face_img)
|
|
if not encodings:
|
|
raise HTTPException(422, detail="Can't find human face")
|
|
res = face_recognition_worker.face_distance(encodings)
|
|
print(res)
|
|
return BaseResponse(result=res)
|
|
|
|
async def main():
|
|
config = uvicorn.Config("server:app", host="0.0.0.0", port=5000, log_level="info")
|
|
server = uvicorn.Server(config)
|
|
await server.serve()
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main()) |