Spaces:
Sleeping
Sleeping
use YOLOv11n
Browse files- models/best_yolov11n.pt +3 -0
- 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 |
-
|
|
|
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/
|
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()}")
|