image-Amiel

Sleeping

kiurtis commited on Feb 5

Commit

5e62440

1 Parent(s): c5ac3e1

hafl precision

Files changed (1) hide show

tasks/image.py CHANGED Viewed

@@ -141,7 +141,7 @@ 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_YOLOv11n_1280_full_batch_1.engine'
     INFERENCE_ENGINE_TYPE = 'pt'
     INPUT_SIZE = 1280
     N_TEST_BATCHES = 2

     #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_YOLOv11n_1280_half_batch_1.engine'
     INFERENCE_ENGINE_TYPE = 'pt'
     INPUT_SIZE = 1280
     N_TEST_BATCHES = 2