Added backup files again

app/Hackathon_setup/exp_recognition_bkp.py

@@ -0,0 +1,71 @@
+import numpy as np
+import cv2
+from matplotlib import pyplot as plt
+import torch
+# In the below line,remove '.' while working on your local system.However Make sure that '.' is present before face_recognition_model while uploading to the server, Do not remove it.
+from .exp_recognition_model import *
+from PIL import Image
+import base64
+import io
+import os
+## Add more imports if required
+
+#############################################################################################################################
+#   Caution: Don't change any of the filenames, function names and definitions                                              #
+#   Always use the current_path + file_name for refering any files, without it we cannot access files on the server         #
+#############################################################################################################################
+
+# Current_path stores absolute path of the file from where it runs.
+current_path = os.path.dirname(os.path.abspath(__file__))
+
+
+#1) The below function is used to detect faces in the given image.
+#2) It returns only one image which has maximum area out of all the detected faces in the photo.
+#3) If no face is detected,then it returns zero(0).
+
+def detected_face(image):
+    eye_haar = current_path + '/haarcascade_eye.xml'
+    face_haar = current_path + '/haarcascade_frontalface_default.xml'
+    face_cascade = cv2.CascadeClassifier(face_haar)
+    eye_cascade = cv2.CascadeClassifier(eye_haar)
+    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+    faces = face_cascade.detectMultiScale(gray, 1.3, 5)
+    face_areas=[]
+    images = []
+    required_image=0
+    for i, (x,y,w,h) in enumerate(faces):
+        face_cropped = gray[y:y+h, x:x+w]
+        face_areas.append(w*h)
+        images.append(face_cropped)
+        required_image = images[np.argmax(face_areas)]
+        required_image = Image.fromarray(required_image)
+    return required_image
+    
+
+#1) Images captured from mobile is passed as parameter to the below function in the API call, It returns the Expression detected by your network.
+#2) The image is passed to the function in base64 encoding, Code for decoding the image is provided within the function.
+#3) Define an object to your network here in the function and load the weight from the trained network, set it in evaluation mode.
+#4) Perform necessary transformations to the input(detected face using the above function), this should return the Expression in string form ex: "Anger"
+#5) For loading your model use the current_path+'your model file name', anyhow detailed example is given in comments to the function 
+##Caution: Don't change the definition or function name; for loading the model use the current_path for path example is given in comments to the function
+def get_expression(img):
+    device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+    
+    ##########################################################################################
+    ##Example for loading a model using weight state dictionary:                            ##
+    ## face_det_net = facExpRec() #Example Network                                          ##
+    ## model = torch.load(current_path + '/exp_recognition_net.t7', map_location=device)    ##
+    ## face_det_net.load_state_dict(model['net_dict'])                                      ##
+    ##                                                                                      ##
+    ##current_path + '/<network_definition>' is path of the saved model if present in       ##
+    ##the same path as this file, we recommend to put in the same directory                 ##
+    ##########################################################################################
+    ##########################################################################################
+
+    face = detected_face(img)
+    if face==0:
+        face = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2GRAY))
+    
+    # YOUR CODE HERE, return expression using your model
+
+    return "YET TO BE CODED"

app/Hackathon_setup/exp_recognition_model_bkp.py

@@ -0,0 +1,31 @@
+import torch
+import torchvision
+import torch.nn as nn
+from torchvision import transforms
+## Add more imports if required
+
+####################################################################################################################
+# Define your model and transform and all necessary helper functions here                                          #
+# They will be imported to the exp_recognition.py file                                                             #
+####################################################################################################################
+
+# Definition of classes as dictionary
+classes = {0: 'ANGER', 1: 'DISGUST', 2: 'FEAR', 3: 'HAPPINESS', 4: 'NEUTRAL', 5: 'SADNESS', 6: 'SURPRISE'}
+
+# Example Network
+class facExpRec(torch.nn.Module):
+    def __init__(self):
+        pass   # remove 'pass' once you have written your code
+        #YOUR CODE HERE
+        
+    def forward(self, x):
+        pass   # remove 'pass' once you have written your code
+        #YOUR CODE HERE
+        
+# Sample Helper function
+def rgb2gray(image):
+    return image.convert('L')
+    
+# Sample Transformation function
+#YOUR CODE HERE for changing the Transformation values.
+trnscm = transforms.Compose([rgb2gray, transforms.Resize((48,48)), transforms.ToTensor()])

app/Hackathon_setup/face_recognition_bkp.py

@@ -0,0 +1,115 @@
+import numpy as np
+import cv2
+from matplotlib import pyplot as plt
+import torch
+# In the below line, remove '.' while working on your local system.
+# Keep '.' before face_recognition_model while uploading to the server
+from .face_recognition_model import *
+from PIL import Image
+import base64
+import io
+import os
+import joblib
+import pickle
+# Add more imports if required
+
+###########################################################################################################################################
+#         Caution: Don't change any of the filenames, function names and definitions                                                      #
+#        Always use the current_path + file_name for refering any files, without it we cannot access files on the server                  # 
+###########################################################################################################################################
+
+# Current_path stores absolute path of the file from where it runs. 
+current_path = os.path.dirname(os.path.abspath(__file__))
+
+# -------------------------
+# Face Detection
+# -------------------------
+def detected_face(image):
+    eye_haar = current_path + '/haarcascade_eye.xml'
+    face_haar = current_path + '/haarcascade_frontalface_default.xml'
+    face_cascade = cv2.CascadeClassifier(face_haar)
+    eye_cascade = cv2.CascadeClassifier(eye_haar)
+    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+    faces = face_cascade.detectMultiScale(gray, 1.3, 5)
+    face_areas = []
+    images = []
+    required_image = 0
+    for i, (x, y, w, h) in enumerate(faces):
+        face_cropped = gray[y:y+h, x:x+w]
+        face_areas.append(w*h)
+        images.append(face_cropped)
+        required_image = images[np.argmax(face_areas)]
+        required_image = Image.fromarray(required_image)
+    return required_image
+
+# -------------------------
+# Compute Similarity
+# -------------------------
+def get_similarity(img1, img2):
+    device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+    
+    # Detect faces
+    det_img1 = detected_face(img1)
+    det_img2 = detected_face(img2)
+    if det_img1 == 0 or det_img2 == 0:
+        det_img1 = Image.fromarray(cv2.cvtColor(img1, cv2.COLOR_BGR2GRAY))
+        det_img2 = Image.fromarray(cv2.cvtColor(img2, cv2.COLOR_BGR2GRAY))
+    
+    # Transform images
+    face1 = trnscm(det_img1).unsqueeze(0).to(device)
+    face2 = trnscm(det_img2).unsqueeze(0).to(device)
+    
+    # -------------------------
+    # Load Siamese Model
+    # -------------------------
+    model_path = current_path + '/siamese_model.t7'
+    checkpoint = torch.load(model_path, map_location=device)
+    feature_net = Siamese().to(device)
+    feature_net.load_state_dict(checkpoint['net_dict'])
+    feature_net.eval()
+    
+    # -------------------------
+    # Compute similarity (Euclidean distance)
+    # -------------------------
+    with torch.no_grad():
+        output1, output2 = feature_net(face1, face2)
+        euclidean_distance = F.pairwise_distance(output1, output2)
+        # Convert distance to similarity score (0–1)
+        similarity_score = 1 / (1 + euclidean_distance.item())
+    
+    return round(similarity_score, 3)
+
+# -------------------------
+# Get Face Class
+# -------------------------
+def get_face_class(img1):
+    device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+    
+    det_img1 = detected_face(img1)
+    if det_img1 == 0:
+        det_img1 = Image.fromarray(cv2.cvtColor(img1, cv2.COLOR_BGR2GRAY))
+    
+    # Transform image
+    face = trnscm(det_img1).unsqueeze(0).to(device)
+    
+    # Load Siamese model
+    model_path = current_path + '/siamese_model.t7'
+    checkpoint = torch.load(model_path, map_location=device)
+    feature_net = Siamese().to(device)
+    feature_net.load_state_dict(checkpoint['net_dict'])
+    feature_net.eval()
+    
+    # -------------------------
+    # Use Siamese + classifier to predict class (if classifier exists)
+    # -------------------------
+    # If you have a trained classifier that takes embeddings from Siamese as input:
+    # classifier = <your classifier>
+    # with torch.no_grad():
+    #     embedding = feature_net.forward_once(face)
+    #     pred = classifier(embedding)
+    #     predicted_class = classes[pred.argmax(dim=1).item()]
+    
+    # Since classifier is not trained here, return placeholder
+    predicted_class = "YET TO BE CODED"
+    
+    return predicted_class

app/Hackathon_setup/face_recognition_model_bkp.py

@@ -0,0 +1,80 @@
+import math
+import torch
+import torchvision
+import torch.nn as nn
+import torch.nn.functional as F
+from torchvision import transforms
+# Add more imports if required
+
+
+# ---------------------------
+# Transformation Function
+# ---------------------------
+# Same transforms as used during training in Colab
+trnscm = transforms.Compose([
+    transforms.Resize((100, 100)),
+    transforms.ToTensor()
+])
+
+
+# ---------------------------
+# Siamese Network Definition
+# ---------------------------
+class Siamese(torch.nn.Module):
+    def __init__(self):
+        super(Siamese, self).__init__()
+        
+        # CNN layers (same as your Colab model)
+        self.cnn1 = nn.Sequential(
+            nn.ReflectionPad2d(1),
+            nn.Conv2d(1, 4, kernel_size=3),
+            nn.ReLU(inplace=True),
+            nn.BatchNorm2d(4),
+
+            nn.ReflectionPad2d(1),
+            nn.Conv2d(4, 8, kernel_size=3),
+            nn.ReLU(inplace=True),
+            nn.BatchNorm2d(8),
+
+            nn.ReflectionPad2d(1),
+            nn.Conv2d(8, 8, kernel_size=3),
+            nn.ReLU(inplace=True),
+            nn.BatchNorm2d(8)
+        )
+
+        # Fully connected layers
+        self.fc1 = nn.Sequential(
+            nn.Linear(8 * 100 * 100, 500),
+            nn.ReLU(inplace=True),
+            nn.Linear(500, 500),
+            nn.ReLU(inplace=True),
+            nn.Linear(500, 5)
+        )
+        
+    def forward_once(self, x):
+        # Forward pass for one image
+        output = self.cnn1(x)
+        output = output.view(output.size()[0], -1)
+        output = self.fc1(output)
+        return output
+        
+    def forward(self, x1, x2):
+        # Forward pass for both images
+        output1 = self.forward_once(x1)
+        output2 = self.forward_once(x2)
+        return output1, output2
+
+
+##########################################################################################################
+## Sample classification network (Specify if you are using a pytorch classifier during the training)    ##
+## classifier = nn.Sequential(nn.Linear(64, 64), nn.BatchNorm1d(64), nn.ReLU(), nn.Linear...)           ##
+##########################################################################################################
+
+# Not used for face similarity — so keep it as None
+classifier = None
+
+
+# ---------------------------
+# Class labels (optional)
+# ---------------------------
+classes = ['person1', 'person2', 'person3', 'person4', 'person5', 'person6', 'person7']

app/Hackathon_setup/team_classifier_bkp.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b9c07b411ab1fa2f2365146b188ea0d4438b5317b88c81b256766c4c5ada3e93
+size 1761

app/main_bkp.py

@@ -0,0 +1,148 @@
+import sys
+from pathlib import Path
+sys.path.append(str(Path(__file__).resolve().parent.parent))
+#print(sys.path)
+from typing import Any
+
+from fastapi import FastAPI, Request, APIRouter, File, UploadFile
+from fastapi.staticfiles import StaticFiles
+from fastapi.templating import Jinja2Templates
+from fastapi.middleware.cors import CORSMiddleware
+from app.config import settings
+from app import __version__
+from app.Hackathon_setup import face_recognition, exp_recognition
+
+import numpy as np
+from PIL import Image
+
+
+app = FastAPI(
+    title=settings.PROJECT_NAME, openapi_url=f"{settings.API_V1_STR}/openapi.json"
+)
+
+# To store files uploaded by users
+app.mount("/static", StaticFiles(directory="app/static"), name="static")
+
+# To access Templates directory
+templates = Jinja2Templates(directory="app/templates")
+
+simi_filename1 = None
+simi_filename2 = None
+face_rec_filename = None
+expr_rec_filename = None
+
+
+####################################  Home Page endpoints  #################################################
+@app.get("/")
+async def root(request: Request):
+    return templates.TemplateResponse("index.html", {'request': request,})
+
+
+####################################  Face Similarity endpoints  #################################################
+@app.get("/similarity/")
+async def similarity_root(request: Request):
+    return templates.TemplateResponse("similarity.html", {'request': request,})
+
+
+@app.post("/predict_similarity/")
+async def create_upload_files(request: Request, file1: UploadFile = File(...), file2: UploadFile = File(...)):
+    global simi_filename1
+    global simi_filename2
+    
+    if 'image' in file1.content_type:
+        contents = await file1.read()
+        simi_filename1 = 'app/static/' + file1.filename
+        with open(simi_filename1, 'wb') as f:
+            f.write(contents)
+    
+    if 'image' in file2.content_type:
+        contents = await file2.read()
+        simi_filename2 = 'app/static/' + file2.filename
+        with open(simi_filename2, 'wb') as f:
+            f.write(contents)
+    
+    img1 = Image.open(simi_filename1)
+    img1 =  np.array(img1).reshape(img1.size[1], img1.size[0], 3).astype(np.uint8)
+    
+    img2 = Image.open(simi_filename2)
+    img2 =  np.array(img2).reshape(img2.size[1], img2.size[0], 3).astype(np.uint8)
+    
+    result = face_recognition.get_similarity(img1, img2)
+    #print(result)
+    
+    return templates.TemplateResponse("predict_similarity.html", {"request": request,
+                                                       "result": np.round(result, 3),
+                                                       "simi_filename1": '../static/'+file1.filename,
+                                                       "simi_filename2": '../static/'+file2.filename,})
+    
+
+####################################  Face Recognition endpoints  #################################################
+@app.get("/face_recognition/")
+async def face_recognition_root(request: Request):
+    return templates.TemplateResponse("face_recognition.html", {'request': request,})
+
+
+@app.post("/predict_face_recognition/")
+async def create_upload_files(request: Request, file3: UploadFile = File(...)):
+    global face_rec_filename
+    
+    if 'image' in file3.content_type:
+        contents = await file3.read()
+        face_rec_filename = 'app/static/' + file3.filename
+        with open(face_rec_filename, 'wb') as f:
+            f.write(contents)
+    
+    img1 = Image.open(face_rec_filename)
+    img1 =  np.array(img1).reshape(img1.size[1], img1.size[0], 3).astype(np.uint8)
+        
+    result = face_recognition.get_face_class(img1)
+    print(result)
+    
+    return templates.TemplateResponse("predict_face_recognition.html", {"request": request,
+                                                       "result": result,
+                                                       "face_rec_filename": '../static/'+file3.filename,})
+
+
+####################################  Expresion Recognition endpoints  #################################################
+@app.get("/expr_recognition/")
+async def expr_recognition_root(request: Request):
+    return templates.TemplateResponse("expr_recognition.html", {'request': request,})
+
+
+@app.post("/predict_expr_recognition/")
+async def create_upload_files(request: Request, file4: UploadFile = File(...)):
+    global expr_rec_filename
+    
+    if 'image' in file4.content_type:
+        contents = await file4.read()
+        expr_rec_filename = 'app/static/' + file4.filename
+        with open(expr_rec_filename, 'wb') as f:
+            f.write(contents)
+    
+    img1 = Image.open(expr_rec_filename)
+    img1 =  np.array(img1).reshape(img1.size[1], img1.size[0], 3).astype(np.uint8)
+        
+    result = exp_recognition.get_expression(img1)
+    print(result)
+    
+    return templates.TemplateResponse("predict_expr_recognition.html", {"request": request,
+                                                       "result": result,
+                                                       "expr_rec_filename": '../static/'+file4.filename,})
+
+
+
+# Set all CORS enabled origins
+if settings.BACKEND_CORS_ORIGINS:
+    app.add_middleware(
+        CORSMiddleware,
+        allow_origins=[str(origin) for origin in settings.BACKEND_CORS_ORIGINS],
+        allow_credentials=True,
+        allow_methods=["*"],
+        allow_headers=["*"],
+    )
+
+
+# Start app
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=8001)

app/templates/expr_recognition_bkp.html

@@ -0,0 +1,32 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <title>Index</title>
+</head>
+<body>
+    <div>
+        <h1 style="background-color:LightGray;">
+            <center>Expression Recognition</center>
+        </h1>
+    </div>
+    <div>
+        <fieldset>
+        <ul>
+            <!li>
+            <br>
+            <form action="/predict_expr_recognition/" enctype="multipart/form-data" method="post">
+                <span style="font-weight:bold;font-family:sans-serif">Upload Image:</span> <br><br>
+                <input name="file4" type="file" onchange="readURL(this);" />
+                <br><br><br>
+                <button type="submit">Recognize Expression</button>
+            </form>
+            <!/li>
+            <br><br>
+            <form action="/" method="get">
+                <button type="submit">Home</button>
+            </form>
+        </ul>
+        </fieldset>
+    </div>
+</body>
+</html>

app/templates/face_recognition_bkp.html

@@ -0,0 +1,32 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <title>Index</title>
+</head>
+<body>
+    <div>
+        <h1 style="background-color:LightGray;">
+            <center>Face Recognition</center>
+        </h1>
+    </div>
+    <div>
+        <fieldset>
+        <ul>
+            <!li>
+            <br>
+            <form action="/predict_face_recognition/" enctype="multipart/form-data" method="post">
+                <span style="font-weight:bold;font-family:sans-serif">Upload Image:</span> <br><br>
+                <input name="file3" type="file" onchange="readURL(this);" />
+                <br><br><br>
+                <button type="submit">Recognize Face</button>
+            </form>
+            <!/li>
+            <br><br>
+            <form action="/" method="get">
+                <button type="submit">Home</button>
+            </form>
+        </ul>
+        </fieldset>
+    </div>
+</body>
+</html>

app/templates/index_bkp.html

@@ -0,0 +1,29 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <title>Index</title>
+</head>
+<body>
+    <div>
+        <h1 style="background-color:LightGray;">
+            <center>Recognition Application</center>
+        </h1>
+    </div>
+    <div>
+        <fieldset>
+        <ul>
+            <li><span style="font-weight:bold;font-family:sans-serif">Select a task:</span> 
+            <br><br><br>
+            <form action="{{ url_for('similarity_root') }}"><button>Face Similarity</button></form>
+            <br><br>
+            <form action="{{ url_for('face_recognition_root') }}"><button>Face Recognition</button></form>
+            <br><br>
+            <form action="{{ url_for('expr_recognition_root') }}"><button>Expression Recognition</button></form>
+            <br>
+            </li>
+            <br>
+        </ul>
+        </fieldset>
+    </div>
+</body>
+</html>

app/templates/predict_expr_recognition_bkp.html

@@ -0,0 +1,37 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <title>Predict</title>
+</head>
+<body>
+    <div>
+        <h1 style="background-color:LightGray;">
+            <center>Expression Recognition</center>
+        </h1>
+    </div>
+    <div>
+        <fieldset>
+            <h2>
+                <center>
+                    <span style="font-weight:bold;font-family:sans-serif">Prediction: </span> 
+                    <span style="font-weight:bold;color:blue"> {{result}}</span>
+                </center>
+            </h2>
+            <h3><center><span style="font-weight:bold;font-family:sans-serif">Input image:</span></Input></center></h3> 
+            <p>
+              <center>
+              <img src="{{expr_rec_filename}}" alt={{expr_rec_filename1}} width='150' height='150'>
+              </center>
+            </p>
+            <br>
+            <form action="/expr_recognition/" method="get">
+                <center><button type="submit">Check Another Input</button></center>
+            </form>
+            <br>
+            <form action="/" method="get">
+                <center><button type="submit">Home</button></center>
+            </form>
+        </fieldset>
+    </div>
+</body>
+</html>

app/templates/predict_face_recognition_bkp.html

@@ -0,0 +1,37 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <title>Predict</title>
+</head>
+<body>
+    <div>
+        <h1 style="background-color:LightGray;">
+            <center>Face Recognition</center>
+        </h1>
+    </div>
+    <div>
+        <fieldset>
+            <h2>
+                <center>
+                    <span style="font-weight:bold;font-family:sans-serif">Prediction: </span> 
+                    <span style="font-weight:bold;color:blue"> {{result}}</span>
+                </center>
+            </h2>
+            <h3><center><span style="font-weight:bold;font-family:sans-serif">Input image:</span></Input></center></h3> 
+            <p>
+              <center>
+              <img src="{{face_rec_filename}}" alt={{face_rec_filename1}} width='150' height='150'>
+              </center>
+            </p>
+            <br>
+            <form action="/face_recognition/" method="get">
+                <center><button type="submit">Check Another Input</button></center>
+            </form>
+            <br>
+            <form action="/" method="get">
+                <center><button type="submit">Home</button></center>
+            </form>
+        </fieldset>
+    </div>
+</body>
+</html>

app/templates/predict_similarity_bkp.html

@@ -0,0 +1,38 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <title>Predict</title>
+</head>
+<body>
+    <div>
+        <h1 style="background-color:LightGray;">
+            <center>Face Similarity</center>
+        </h1>
+    </div>
+    <div>
+        <fieldset>
+            <h2>
+                <center>
+                    <span style="font-weight:bold;font-family:sans-serif">Dissimilarity:</span> 
+                    <span style="font-weight:bold;color:blue"> {{result}}</span>
+                </center>
+            </h2>
+            <h3><center><span style="font-weight:bold;font-family:sans-serif">Input images:</span></Input></center></h3> 
+            <p>
+              <center>
+              <img src="{{simi_filename1}}" alt={{simi_filename1}} width='150' height='150'>
+              <img src="{{simi_filename2}}" alt={{simi_filename2}} width='150' height='150'>
+              </center>
+            </p>
+            <br>
+            <form action="/similarity/" method="get">
+                <center><button type="submit">Check Another Input</button></center>
+            </form>
+            <br>
+            <form action="/" method="get">
+                <center><button type="submit">Home</button></center>
+            </form>
+        </fieldset>
+    </div>
+</body>
+</html>

app/templates/similarity_bkp.html

@@ -0,0 +1,35 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <title>Index</title>
+</head>
+<body>
+    <div>
+        <h1 style="background-color:LightGray;">
+            <center>Face Similarity</center>
+        </h1>
+    </div>
+    <div>
+        <fieldset>
+        <ul>
+            <!li>
+            <br>
+            <form action="/predict_similarity/" enctype="multipart/form-data" method="post">
+                <span style="font-weight:bold;font-family:sans-serif">Upload First Image:</span> <br><br>
+                <input name="file1" type="file" onchange="readURL(this);" />
+                <br><br><br>
+                <span style="font-weight:bold;font-family:sans-serif">Upload Second Image:</span> <br><br>
+                <input name="file2" type="file" onchange="readURL(this);" />
+                <br><br><br><br>
+                <button type="submit">Check Similarity</button>
+            </form>
+            <!/li>
+            <br><br>
+            <form action="/" method="get">
+                <button type="submit">Home</button>
+            </form>
+        </ul>
+        </fieldset>
+    </div>
+</body>
+</html>