kiurtis commited on
Commit
37c12d6
·
1 Parent(s): 9ebd3cc

feat(models): add quantisized YOLOv11 1280

Browse files
models/best_YOLOv11n_1280_real_half.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a40907b815e2d13b8834f79128a72491f8d73e3e9840d3ad39f4f3555023ea9e
3
+ size 5567977
tasks/image.py CHANGED
@@ -16,7 +16,7 @@ load_dotenv()
16
  router = APIRouter()
17
 
18
  #MODEL_TYPE = "YOLOv11n"
19
- DESCRIPTION = f"Simple YOLOv6n model 1280"
20
  ROUTE = "/image"
21
 
22
  def parse_boxes(annotation_string):
@@ -117,7 +117,8 @@ async def evaluate_image(request: ImageEvaluationRequest):
117
  import matplotlib.pyplot as plt
118
 
119
  #PATH_TO_MODEL = 'models/best_YOLOv11n_1280.onnx'
120
- PATH_TO_MODEL = 'models/best_yolov6n_1280.pt'
 
121
  INFERENCE_ENGINE_TYPE = 'pt'
122
  INPUT_SIZE = 1280
123
 
 
16
  router = APIRouter()
17
 
18
  #MODEL_TYPE = "YOLOv11n"
19
+ DESCRIPTION = f"YOLOv11n model 1280 with half precision"
20
  ROUTE = "/image"
21
 
22
  def parse_boxes(annotation_string):
 
117
  import matplotlib.pyplot as plt
118
 
119
  #PATH_TO_MODEL = 'models/best_YOLOv11n_1280.onnx'
120
+ #PATH_TO_MODEL = 'models/best_yolov6n_1280.pt'
121
+ PATH_TO_MODEL = 'models/best_YOLOv11n_1280_real_half.onnx'
122
  INFERENCE_ENGINE_TYPE = 'pt'
123
  INPUT_SIZE = 1280
124