Hugging Face's logo

Spaces:
frugal-ai-challenge
/
image-Amiel
Sleeping

App Files Community
kiurtis commited on
Commit
8d8b751
·
1 Parent(s): c7db377

with quantisized models

Browse files
Files changed (2) hide show
  1. models/best_YOLOv8n.onnx +3 -0
  2. tasks/image.py +1 -1
models/best_YOLOv8n.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:605f7f7dd4e3f5fcfddcdfe2863bf09c5f3b322259e8dc99026f8e2b5237c978
3
+ size 6142609
tasks/image.py CHANGED
@@ -120,7 +120,7 @@ async def evaluate_image(request: ImageEvaluationRequest):
120
  #PATH_TO_MODEL = 'models/best_YOLOv11n_1280.onnx'
121
  #PATH_TO_MODEL = 'models/best_yolov6n_1280.pt'
122
  #PATH_TO_MODEL = 'models/best_YOLOv11n_1280_real_half.onnx'
123
- PATH_TO_MODEL = 'models/best_YOLOv8n.pt'
124
  INFERENCE_ENGINE_TYPE = 'pt'
125
  INPUT_SIZE = 640
126
  N_TEST_BATCHES = 2
 
120
  #PATH_TO_MODEL = 'models/best_YOLOv11n_1280.onnx'
121
  #PATH_TO_MODEL = 'models/best_yolov6n_1280.pt'
122
  #PATH_TO_MODEL = 'models/best_YOLOv11n_1280_real_half.onnx'
123
+ PATH_TO_MODEL = 'models/best_YOLOv8n.onnx'
124
  INFERENCE_ENGINE_TYPE = 'pt'
125
  INPUT_SIZE = 640
126
  N_TEST_BATCHES = 2