CoolFace
Datasetpublic

crystal-technologies/CircumSpect

sourceHugging Faceupdated 2y agoView on Hugging Face
0likes232downloads
analyze.py32 linesDownload Raw Back to root
1from dense import describe_image2from faces import recognize_users3import cv24import time5 6def analyze_image(cap):7    users, recognized = recognize_users(cap)8 9    captions, annotated = describe_image(recognized)10 11    if users == []:12        users = "No faces identified"13 14    if type(users) == list:15        users = ", ".join(users)16 17    output = f"""Faces: {users}18View description: {", ".join(captions)}"""19    return output, annotated20 21# cap = cv2.VideoCapture(0)22# while True:23#     output, image = analyze_image(cap)24#     print(output)25 26#     cv2.imshow("Annotated Image", image)27#     if cv2.waitKey(1) & 0xFF == ord('q'):28#         break29 30# cap.release()31# cv2.destroyAllWindows()32