Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import os | |
| import keras | |
| from keras.preprocessing.image import img_to_array | |
| import imutils | |
| import cv2 | |
| from keras.models import load_model | |
| import numpy as np | |
| # parameters for loading data and images | |
| detection_model_path = 'haarcascade_files/haarcascade_frontalface_default.xml' | |
| # emotion_model_path = 'model2/model2_entire_model.h5' | |
| emotion_model_path = 'model_2_aug_nocall_BEST/model_2_aug_nocall_entire_model.h5' | |
| # hyper-parameters for bounding boxes shape | |
| # loading models | |
| face_detection = cv2.CascadeClassifier(detection_model_path) | |
| emotion_classifier = load_model(emotion_model_path, compile=False) | |
| EMOTIONS = ['neutral','happiness','surprise','sadness','anger','disgust','fear','contempt','unknown'] | |
| def predict(frame): | |
| frame = imutils.resize(frame, width=300) | |
| gray = cv2.cvtColor(frame, cv2.COLOR_RGB2GRAY) | |
| faces = face_detection.detectMultiScale(gray, scaleFactor=1.1, | |
| minNeighbors=5, minSize=(30, 30), | |
| flags=cv2.CASCADE_SCALE_IMAGE) | |
| frameClone = frame.copy() | |
| if len(faces) > 0: | |
| faces = sorted(faces, reverse=True, | |
| key=lambda x: (x[2] - x[0]) * (x[3] - x[1]))[0] | |
| (fX, fY, fW, fH) = faces | |
| # Extract the ROI of the face from the grayscale image, resize it to a fixed 28x28 pixels, and then prepare | |
| # the ROI for classification via the CNN | |
| roi = gray[fY:fY + fH, fX:fX + fW] | |
| roi = cv2.resize(roi, (48, 48)) | |
| roi = roi.astype("float") / 255.0 | |
| roi = img_to_array(roi) | |
| roi = np.expand_dims(roi, axis=0) | |
| preds = emotion_classifier.predict(roi)[0] | |
| label = EMOTIONS[preds.argmax()] | |
| else: | |
| return frameClone, "Can't find your face" | |
| probs = {} | |
| cv2.putText(frameClone, label, (fX, fY - 10), | |
| cv2.FONT_HERSHEY_DUPLEX, 1, (238, 164, 64), 1) | |
| cv2.rectangle(frameClone, (fX, fY), (fX + fW, fY + fH), | |
| (238, 164, 64), 2) | |
| for (i, (emotion, prob)) in enumerate(zip(EMOTIONS, preds)): | |
| probs[emotion] = float(prob) | |
| return frameClone, probs | |
| # Define Gradio input and output components | |
| image_input = gr.components.Image(type='numpy', label="Upload Image or Video") | |
| image_output = gr.components.Image(label="Predicted Emotion") | |
| label_output = gr.components.Label(num_top_classes=2, label="Top 2 Probabilities") | |
| inp = [ | |
| gr.components.Image(sources="webcam", label="Your face"), | |
| gr.components.File(label="Upload Image or Video") | |
| ] | |
| out = [ | |
| gr.components.Image(label="Predicted Emotion"), | |
| gr.components.Label(num_top_classes=2, label="Top 2 Probabilities") | |
| ] | |
| example_images = [ | |
| [ | |
| os.path.join(os.path.dirname(__file__), "images/chandler.jpeg"), | |
| os.path.join(os.path.dirname(__file__), "images/janice.jpg"), | |
| os.path.join(os.path.dirname(__file__), "images/joey.jpg"), | |
| os.path.join(os.path.dirname(__file__), "images/phoebe.jpg"), | |
| os.path.join(os.path.dirname(__file__), "images/rachel_monica.jpg"), | |
| os.path.join(os.path.dirname(__file__), "images/ross.jpg"), | |
| os.path.join(os.path.dirname(__file__), "images/gunther.jpg") | |
| ] | |
| ] | |
| # example_images = [ | |
| # ["images/chandler.jpeg"], | |
| # ["images/janice.jpg"], | |
| # ["images/joey.jpg"], | |
| # ["images/phoebe.jpg"], | |
| # ["images/rachel_monica.jpg"], | |
| # ["images/ross.jpg"], | |
| # ["images/gunther.jpg"] | |
| # ] | |
| title = "Facial Emotion Recognition" | |
| description = "How well can this model predict your emotions? Take a picture with your webcam, and it will guess if" \ | |
| " you are: happy, sad, angry, disgusted, scared, surprised, or neutral." | |
| thumbnail = "https://raw.githubusercontent.com/gradio-app/hub-emotion-recognition/master/thumbnail.png" | |
| # gr.Interface(predict, inp, out, capture_session=True, title=title, thumbnail=thumbnail, | |
| # description=description).launch(inbrowser=True) | |
| gr.Interface(fn=predict, inputs=inp, outputs=out, | |
| examples=example_images,title=title, thumbnail=thumbnail).launch() | |
| # # Launch Gradio interface | |
| # gr.Interface(fn=predict, inputs=image_input, outputs=[image_output, label_output], | |
| # title="Facial Emotion Recognition", description="How well can this model predict your emotions?").launch() | |