kiurtis commited on
Commit
93d080a
·
1 Parent(s): 51fe6d6

use YOLOv11n

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