| from yolo.yoloFace import YOLO_FACE | |
| from vgg.vgg_face import MODEL_FACE | |
| from database.retriever import BruteForceStore | |
| import cv2 | |
| DB = BruteForceStore() | |
| def pipeline(img): | |
| images = YOLO_FACE(img) | |
| for patch in images: | |
| embeddings = MODEL_FACE(patch) | |
| if DB(embeddings): return "Welcome!" | |
| return "Buzz off!!! petrichor me chor" | |
| if __name__ == "__main__": | |
| import cv2 | |
| img = cv2.imread('temp.jpg', cv2.IMREAD_UNCHANGED) | |
| print(pipeline(img)) |