Spaces:
Sleeping
Sleeping
feat(model): YOLOv11n 1280
Browse files- models/best_YOLOv11n_1280.pt +3 -0
- tasks/image.py +2 -2
models/best_YOLOv11n_1280.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3eef8c404266d3dbde4aa9aa8a0a63bdd98eb7b4b3880f5240d5b6ff0a9bc414
|
| 3 |
+
size 5540627
|
tasks/image.py
CHANGED
|
@@ -16,7 +16,7 @@ load_dotenv()
|
|
| 16 |
router = APIRouter()
|
| 17 |
|
| 18 |
#MODEL_TYPE = "YOLOv11n"
|
| 19 |
-
DESCRIPTION = f"Simple
|
| 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/
|
| 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 YOLOv11n model 1280"
|
| 20 |
ROUTE = "/image"
|
| 21 |
|
| 22 |
def parse_boxes(annotation_string):
|
|
|
|
| 113 |
#--------------------------------------------------------------------------------------------
|
| 114 |
|
| 115 |
|
| 116 |
+
PATH_TO_MODEL = f"models/best_YOLOv11n_1280.pt"
|
| 117 |
model = load_model(PATH_TO_MODEL)
|
| 118 |
|
| 119 |
print(f"Model info: {model.info()}")
|