Spaces:
Sleeping
Sleeping
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"
|
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 |
|