tensorrt yolov11n model 32 batch

models/best_YOLOv11n_1280_full_batch_32.engine

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

tasks/image.py

@@ -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_128.engine'
+    PATH_TO_MODEL = 'models/best_YOLOv11n_1280_full_batch_32.engine'
     INFERENCE_ENGINE_TYPE = 'pt'
-    INPUT_SIZE = 640
+    INPUT_SIZE = 1280
     N_TEST_BATCHES = 2
-    BATCH_SIZE = 128 # Can be adjusted as needed
+    BATCH_SIZE = 32 # 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