ExecuTorch optimization, UI improvements..

.gitattributes

@@ -21,6 +21,7 @@
 *.pickle filter=lfs diff=lfs merge=lfs -text
 *.pkl filter=lfs diff=lfs merge=lfs -text
 *.pt filter=lfs diff=lfs merge=lfs -text
+*.pte filter=lfs diff=lfs merge=lfs -text
 *.pth filter=lfs diff=lfs merge=lfs -text
 *.rar filter=lfs diff=lfs merge=lfs -text
 *.safetensors filter=lfs diff=lfs merge=lfs -text

README.md

@@ -18,7 +18,7 @@ Full-stack traffic analytics dashboard. YOLO + ByteTrack backend processes uploa
 
 ## Stack
 
-- **Backend**: FastAPI, Ultralytics YOLO (OpenVINO), ByteTrack, OpenCV
+- **Backend**: FastAPI, Ultralytics YOLO (ExecuTorch), ByteTrack, OpenCV
 - **Frontend**: Vanilla HTML/JS, TailwindCSS CDN, Chart.js
 - **Infra**: Docker, CPU-only inference
 

backend/model.py

@@ -7,10 +7,10 @@ load_dotenv()
 
 MODEL_DIR = Path(__file__).parent / "weights"
 PT_PATH = MODEL_DIR / "best.pt"
-OV_DIR = MODEL_DIR / "best_int8_openvino_model"
+ET_DIR = MODEL_DIR / "yolo26n_executorch_model"
 
 
-def ensure_openvino():
+def ensure_executorch():
     MODEL_DIR.mkdir(exist_ok=True)
 
     if not PT_PATH.exists():
@@ -21,11 +21,11 @@ def ensure_openvino():
             f'https://huggingface.co/Perception365/VehicleNet-Y26s/resolve/main/weights/best.pt'
         )
 
-    # https://docs.ultralytics.com/modes/export/#arguments
-    if not OV_DIR.exists():
-        YOLO(str(PT_PATH)).export(format="openvino", dynamic=True, int8=True)
+    # https://docs.ultralytics.com/integrations/executorch/
+    if not ET_DIR.exists():
+        YOLO(str(PT_PATH)).export(format="executorch", device='cpu')
 
 
 def load_model():
-    ensure_openvino()
-    return YOLO(str(OV_DIR), task="detect")
+    ensure_executorch()
+    return YOLO(str(ET_DIR), task="detect")

frontend/initial.html

@@ -85,7 +85,7 @@
             </h1>
             <p class="font-bold mb-8 text-sm uppercase tracking-[0.2em] text-neutral-500 flex items-center">
                 <span class="core-badge px-3 py-1 rounded-full text-[10px] mr-3">v1.0 CORE</span>
-                Powered by Deep Learning Inference
+                Powered by Deep Learning 
             </p>
             <ul class="space-y-4 xl:space-y-5 text-base xl:text-lg font-medium text-neutral-400">
                 <li class="flex items-center"><i class="fa-solid fa-check text-white mr-5 text-xl"></i> Real-time
@@ -107,7 +107,7 @@
             <!-- STEP: Modules -->
             <div id="step-modules" class="w-full flex flex-col fade-in">
                 <h2 class="text-3xl font-bold mb-2 text-white text-center">UrbanFlow</h2>
-                <p class="text-[13px] font-medium text-neutral-500 mb-8 text-center">Select an analytical pipeline to
+                <p class="text-[13px] font-medium text-neutral-500 mb-8 text-center">Select below analytical pipeline to
                     proceed.</p>
 
                 <div class="flex justify-center w-full">
@@ -115,7 +115,7 @@
                         class="group relative border-2 border-white rounded-[2rem] p-8 cursor-pointer hover:-translate-y-1 transition-all duration-300 text-center max-w-sm w-full traffic-dynamics-card">
                         <div
                             class="absolute top-4 right-6 bg-white text-black text-[9px] font-bold px-2.5 py-1 rounded-full uppercase tracking-wider">
-                            Active</div>
+                            V1.0</div>
                         <i class="fa-solid fa-car-side text-4xl text-white mb-4 block mx-auto"></i>
                         <h3 class="font-bold text-lg mb-2 leading-tight text-white">Traffic <br>Dynamics</h3>
                         <p class="text-[10px] text-neutral-500 font-medium leading-relaxed">Detect, track, and analyze
@@ -128,11 +128,10 @@
             <div id="step-upload" class="hidden w-full flex flex-col fade-in">
                 <button onclick="showStep('modules')"
                     class="text-neutral-500 hover:text-white transition flex items-center text-xs font-bold uppercase tracking-widest mb-6 w-fit">
-                    <i class="fa-solid fa-arrow-left mr-2"></i> Back to Suite
+                    <i class="fa-solid fa-arrow-left mr-2"></i> Back 
                 </button>
-                <h2 class="text-3xl font-bold mb-2 text-white text-center">Define Media Source</h2>
-                <p class="text-[13px] font-medium text-neutral-500 mb-8 text-center">Provide the target video footage to
-                    initialize the pipeline.</p>
+                <h2 class="text-3xl font-bold mb-2 text-white text-center">Define Media</h2>
+                <p class="text-[13px] font-medium text-neutral-500 mb-8 text-center">Provide a target video footage. </p>
 
                 <input id="file-input" type="file" accept="video/*" class="hidden">
                 <div id="dropzone" onclick="document.getElementById('file-input').click()"
@@ -160,8 +159,8 @@
             <!-- STEP: Draw -->
             <div id="step-draw" class="hidden w-full flex flex-col fade-in">
                 <h2 class="text-3xl font-bold mb-2 text-white text-center">Define Boundary</h2>
-                <p class="text-[11px] font-bold uppercase tracking-widest text-neutral-500 mb-6 text-center">Establish
-                    counting threshold points</p>
+                <p class="text-[11px] font-bold uppercase tracking-widest text-neutral-500 mb-6 text-center">Mark two
+                    points to define counting threshold</p>
 
                 <div
                     class="relative w-full aspect-video bg-neutral-950 rounded-3xl overflow-hidden cursor-crosshair mb-6">
@@ -179,7 +178,7 @@
                     <button onclick="resetCanvas()"
                         class="w-1/3 py-3.5 bg-neutral-900 border border-neutral-800 text-white rounded-full font-bold hover:bg-neutral-800 transition-all text-sm">Reset</button>
                     <button id="btn-proceed" onclick="startRun()"
-                        class="w-2/3 py-3.5 bg-[#666666] text-black rounded-full font-bold transition-all text-center text-sm">Continue
+                        class="w-2/3 py-3.5 bg-white text-black rounded-full font-bold transition-all text-center text-sm">Continue
                         &nbsp;&rarr;</button>
                 </div>
             </div>

frontend/vehicles.html

@@ -421,7 +421,7 @@
                             <ul class="text-xs space-y-3 pl-1">
                                 <li class="flex items-start gap-3"><i
                                         class="fa-solid fa-circle text-[5px] mt-1.5 text-slate-400"></i>
-                                    <span>OpenVINO Performance Optimization</span>
+                                    <span>ExecuTorch Performance Optimization</span>
                                 </li>
                                 <li class="flex items-start gap-3"><i
                                         class="fa-solid fa-circle text-[5px] mt-1.5 text-slate-400"></i>
@@ -574,7 +574,7 @@
                 </div>
 
                 <!-- Technical Context Row -->
-                <div class="grid grid-cols-3 gap-6">
+                <div class="grid grid-cols-2 gap-6">
                     <div class="bg-white rounded-xl border border-slate-200 shadow-sm flex flex-col overflow-hidden">
                         <div class="px-6 py-4 border-b border-slate-100 bg-slate-50/50">
                             <h3 class="font-bold text-slate-800 text-sm">Stream Source Profile</h3>
@@ -589,10 +589,16 @@
                     </div>
                     <div class="bg-white rounded-xl border border-slate-200 shadow-sm flex flex-col overflow-hidden">
                         <div class="px-6 py-4 border-b border-slate-100 bg-slate-50/50">
-                            <h3 class="font-bold text-slate-800 text-sm">Model Architecture & Logic</h3>
+                            <h3 class="font-bold text-slate-800 text-sm">Model Architecture &amp; Logic</h3>
                         </div>
                         <div class="p-6 space-y-4" id="panel-model"></div>
                     </div>
+                    <div class="bg-white rounded-xl border border-slate-200 shadow-sm flex flex-col overflow-hidden">
+                        <div class="px-6 py-4 border-b border-slate-100 bg-slate-50/50">
+                            <h3 class="font-bold text-slate-800 text-sm">Inference Parameters</h3>
+                        </div>
+                        <div class="p-6 space-y-4" id="panel-infer"></div>
+                    </div>
                 </div>
 
             </div>
@@ -882,14 +888,18 @@
                     <a href="https://huggingface.co/Perception365/VehicleNet-Y26s" target="_blank" class="text-sm font-bold text-white mono-font hover:text-slate-300 transition underline underline-offset-4 decoration-slate-700">Perception365/VehicleNet-Y26s</a>
                 </div>` +
                 detailRow('task', 'detect') +
-                detailRow('format', 'OpenVINO') +
+                detailRow('format', 'ExecuTorch') +
                 detailRow('tracker', 'ByteTrack');
 
+            populateInferPanel(c);
+        }
+
+        function populateInferPanel(c) {
             document.getElementById('panel-infer').innerHTML =
                 infoRow('imgsz', c.imgsz, 'Input image resolution for model inference.') +
                 infoRow('vid_stride', c.detect_stride, 'Frames skipped between consecutive detections relative to the spatial boundary.') +
-                infoRow('conf', c.conf || 0.12, 'Minimum confidence threshold for valid detections.') +
-                infoRow('iou', c.iou || 0.60, 'Intersection-over-Union threshold for non-max suppression.') +
+                infoRow('conf', (c.conf || 0.12).toFixed(2), 'Minimum confidence threshold for valid detections.') +
+                infoRow('iou', (c.iou || 0.60).toFixed(2), 'Intersection-over-Union threshold for non-max suppression.') +
                 infoRow('stream', 'TRUE', 'Frame-by-frame processing for constant memory usage.') +
                 infoRow('verbose', 'FALSE', 'Console logging suppressed during inference.');
         }
@@ -1218,6 +1228,9 @@
             _params.config.iou = iou;
             _params.config.detect_stride = stride;
 
+            // Reflect final resolved params in Run tab
+            populateInferPanel(_params.config);
+
             // Lock settings
             lockSettings();
 

requirements.txt

@@ -7,10 +7,7 @@ ultralytics>=8.3.0
 numpy==1.24.3
 python-dotenv==1.0.0
 websockets==12.0
-onnxruntime
-openvino>=2024.0.0
-onnx>=1.12.0
-onnxslim>=0.1.71
+executorch
 lap>=0.5.12
 torch==2.1.0+cpu
 torchvision==0.16.0+cpu