Merge branch 'main' of hf.co:spaces/Amiel/terra-cognita-frugal-ai

.gitattributes

@@ -35,4 +35,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 models/best_YOLOv8n_full_batch_1.engine filter=lfs diff=lfs merge=lfs -text
 models/best_YOLOv8n_half_batch_64.engine filter=lfs diff=lfs merge=lfs -text
+models/best_YOLOv8n_full_batch_64.engine filter=lfs diff=lfs merge=lfs -text
+models/best_YOLOv8n_half_batch_128.engine filter=lfs diff=lfs merge=lfs -text
+models/best_YOLOv8n_full_batch_128.engine filter=lfs diff=lfs merge=lfs -text
 *.engine filter=lfs diff=lfs merge=lfs -text

models/best_YOLOv8n_full_batch_128.engine

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

models/best_YOLOv8n_full_batch_64.engine

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

models/best_YOLOv8n_half_batch_128.engine

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

tasks/image.py

@@ -37,7 +37,7 @@ print("CUDA version:", torch.version.cuda)
 print("cuDNN version:", torch.backends.cudnn.version())
 
 #MODEL_TYPE = "YOLOv11n"
-DESCRIPTION = f"YOLOv8n model with batch 64 inference on TensorRT"
+DESCRIPTION = f"YOLOv8n quantisized model with batch 64 inference on TensorRT"
 print(14)
 ROUTE = "/image"
 print(15)
@@ -141,11 +141,11 @@ async def evaluate_image(request: ImageEvaluationRequest):
     #PATH_TO_MODEL = 'models/best_YOLOv11n_1280.onnx'
     #PATH_TO_MODEL = 'models/best_yolov6n_1280.pt'
     #PATH_TO_MODEL = 'models/best_YOLOv11n_1280_real_half.onnx'
-    PATH_TO_MODEL = 'models/best_YOLOv8n_full_batch_64.engine'
+    PATH_TO_MODEL = 'models/best_YOLOv8n_full_batch_128.engine'
     INFERENCE_ENGINE_TYPE = 'pt'
     INPUT_SIZE = 640
     N_TEST_BATCHES = 2
-    BATCH_SIZE = 64 # Can be adjusted as needed
+    BATCH_SIZE = 128 # Can be adjusted as needed
     print("PATH_TO_MODEL", PATH_TO_MODEL)
     def preprocessor(frame):
         #frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) # Only when read from file
@@ -309,7 +309,7 @@ async def evaluate_image(request: ImageEvaluationRequest):
             print(n_box_distr)
             #break
             pass
-        pred_boxes = []
+    pred_boxes = []
 
     for idx in range(len(all_preds)):
         if has_smoke_list[idx]: