Initial commit

app.py

@@ -0,0 +1,73 @@
+import os
+import sys
+import cv2
+import numpy as np
+import pandas as pd
+import streamlit as st
+import joblib
+
+SCALER_PATH = r"C:\Users\Kumar\Desktop\Driver-pose-detection\scaler.pkl"
+ENCODER_PATH = r"C:\Users\Kumar\Desktop\Driver-pose-detection\le.pkl"
+MODEL_PATH = r"C:\Users\Kumar\Desktop\Driver-pose-detection\randomforest.pkl"
+
+scaler = joblib.load(SCALER_PATH)
+encoder = joblib.load(ENCODER_PATH)
+model = joblib.load(MODEL_PATH)
+pred = None
+def live_data_pred(cam_input):
+    bytes_data = cam_input.getvalue()
+
+    cv2_img = cv2.imdecode(np.frombuffer(bytes_data, np.uint8), 0)
+    resize_img = cv2.resize(cv2_img, (256, 256), interpolation=cv2.INTER_CUBIC)
+    flattened_img = resize_img.flatten()
+    flattened_img = np.reshape(flattened_img, (1, -1))
+    
+    scaled_input = scaler.transform(flattened_img)
+    prediction = model.predict(scaled_input)
+    prediction = encoder.inverse_transform(prediction)
+
+    return prediction
+
+def pick_gif(prediction):
+    pass
+
+
+
+
+st.set_page_config(page_title="Driver Posture Alert System", layout='wide')
+st.markdown("## Driver Posture Detection and Alert System")
+st.markdown(">A user friendly platform Indepedent application to detect the real time posture of the driver and provides appropriate actions")
+
+with st.sidebar:
+    st.markdown("#### Choose the Input type")
+    opt = st.selectbox(
+        "Select the type of input",
+        ("Live camera input", "Captured input"),
+        index = None
+    )
+    
+    if opt == "Live camera input":
+        cam_input = st.camera_input("Live input")
+        if cam_input is not None:
+            pred = live_data_pred(cam_input)
+            st.write(f"{pred}")
+            #gif_path = pick_gif(pred)
+
+    elif opt == "Captured input":
+        cam_input = st.file_uploader(
+            label="Upload captured image",
+            type= ["png", "jpg", "jpeg"]
+        )
+        if cam_input is not None:
+            pred = live_data_pred(cam_input)
+            
+
+if pred is not None:
+    st.markdown("### Image Captured - Prediction")
+    st.write(f"{pred[0].capitalize()}")
+else:
+    st.markdown("#### Waiting for Input...")
+
+        
+
+

le.pkl

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

randomforest.pkl

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

requirements.txt

@@ -0,0 +1,6 @@
+joblib
+pandas
+numpy
+scikit-learn
+opencv-python
+streamlit

scaler.pkl

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