kiurtis commited on
Commit
5ed9537
·
1 Parent(s): c46af6f

feat(model): add yolo v11

Browse files
Files changed (2) hide show
  1. models/best_yolov11l.pt +3 -0
  2. tasks/image.py +2 -2
models/best_yolov11l.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0566d85da58da05e4ccb37fe3b96df5ea01eda03acbb96d17195506ae930309a
3
+ size 51176594
tasks/image.py CHANGED
@@ -16,7 +16,7 @@ load_dotenv()
16
  router = APIRouter()
17
 
18
  #MODEL_TYPE = "YOLOv11n"
19
- DESCRIPTION = f"Simple YOLOv11n model"
20
  ROUTE = "/image"
21
 
22
  def parse_boxes(annotation_string):
@@ -113,7 +113,7 @@ async def evaluate_image(request: ImageEvaluationRequest):
113
  #--------------------------------------------------------------------------------------------
114
 
115
 
116
- PATH_TO_MODEL = f"models/best_YOLOv11n.pt"
117
  model = load_model(PATH_TO_MODEL)
118
 
119
  print(f"Model info: {model.info()}")
 
16
  router = APIRouter()
17
 
18
  #MODEL_TYPE = "YOLOv11n"
19
+ DESCRIPTION = f"Simple YOLOv11l model"
20
  ROUTE = "/image"
21
 
22
  def parse_boxes(annotation_string):
 
113
  #--------------------------------------------------------------------------------------------
114
 
115
 
116
+ PATH_TO_MODEL = f"models/best_YOLOv11l.pt"
117
  model = load_model(PATH_TO_MODEL)
118
 
119
  print(f"Model info: {model.info()}")