Skip to content

Commit 716a9ab

Browse files
authored
Yolo26 (#1914)
* Add yolo26 * Isort * Add yolo26 tests * Bugfix * Bump version * isort * License * License * Fix format for yolo26 objdet model * style * Add yolo26 methods to codeflash ignore bc test is taking forever * Fix ignore * Remove redundant ops * Style
1 parent 77b91b3 commit 716a9ab

16 files changed

+4774
-2
lines changed

inference/core/models/object_detection_base.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -122,6 +122,7 @@ def make_response(
122122
predictions = predictions[
123123
: len(img_dims)
124124
] # If the batch size was fixed we have empty preds at the end
125+
125126
responses = [
126127
ObjectDetectionInferenceResponse(
127128
predictions=[

inference/core/version.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
__version__ = "0.63.5"
1+
__version__ = "0.64.0"
22

33

44
if __name__ == "__main__":

inference/models/README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ The models supported by Roboflow Inference have their own licenses. View the lic
3030
| `inference/models/yolov9` | [GPL-3.0](https://github.com/WongKinYiu/yolov9/blob/main/LICENSE.md) ||
3131
| `inference/models/yolov10` | [AGPL-3.0](https://github.com/THU-MIG/yolov10/blob/main/LICENSE) ||
3232
| `inference/models/yolov11` | [AGPL-3.0](https://github.com/ultralytics/ultralytics/blob/master/LICENSE) ||
33+
| `inference/models/yolo26` | [AGPL-3.0](https://github.com/ultralytics/ultralytics/blob/master/LICENSE) ||
3334
| `inference/models/yolov12` | [AGPL-3.0](https://github.com/sunsmarterjie/yolov12?tab=AGPL-3.0-1-ov-file) ||
3435
| `inference/models/smolvlm2` | [Apache 2.0](https://huggingface.co/HuggingFaceTB/SmolVLM2-2.2B-Instruct) | 👍 |
3536
| `inference/models/depth_estimation` | [Apache 2.0](https://huggingface.co/depth-anything/Depth-Anything-V2-Small) | 👍 |

inference/models/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -100,6 +100,9 @@
100100
"YOLOv11KeypointsDetection": "inference.models.yolov11",
101101
"YOLOv11ObjectDetection": "inference.models.yolov11",
102102
"YOLOv12ObjectDetection": "inference.models.yolov12",
103+
"YOLO26InstanceSegmentation": "inference.models.yolo26",
104+
"YOLO26KeypointsDetection": "inference.models.yolo26",
105+
"YOLO26ObjectDetection": "inference.models.yolo26",
103106
}
104107

105108

inference/models/utils.py

Lines changed: 60 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,8 @@
4040
RFDETRInstanceSegmentation,
4141
RFDETRObjectDetection,
4242
VitClassification,
43+
YOLO26InstanceSegmentation,
44+
YOLO26ObjectDetection,
4345
YOLONASObjectDetection,
4446
YOLOv5InstanceSegmentation,
4547
YOLOv5ObjectDetection,
@@ -53,6 +55,7 @@
5355
YOLOv11ObjectDetection,
5456
YOLOv12ObjectDetection,
5557
)
58+
from inference.models.yolo26.yolo26_keypoints_detection import YOLO26KeypointsDetection
5659
from inference.models.yolov8.yolov8_keypoints_detection import YOLOv8KeypointsDetection
5760
from inference.models.yolov11.yolov11_keypoints_detection import (
5861
YOLOv11KeypointsDetection,
@@ -111,6 +114,13 @@
111114
("object-detection", "yolov12m"): YOLOv12ObjectDetection,
112115
("object-detection", "yolov12l"): YOLOv12ObjectDetection,
113116
("object-detection", "yolov12x"): YOLOv12ObjectDetection,
117+
("object-detection", "yolo26"): YOLO26ObjectDetection,
118+
("object-detection", "yolo26s"): YOLO26ObjectDetection,
119+
("object-detection", "yolo26n"): YOLO26ObjectDetection,
120+
("object-detection", "yolo26b"): YOLO26ObjectDetection,
121+
("object-detection", "yolo26m"): YOLO26ObjectDetection,
122+
("object-detection", "yolo26l"): YOLO26ObjectDetection,
123+
("object-detection", "yolo26x"): YOLO26ObjectDetection,
114124
("object-detection", "rfdetr-base"): RFDETRObjectDetection,
115125
("object-detection", "rfdetr-large"): RFDETRObjectDetection,
116126
("object-detection", "rfdetr-nano"): RFDETRObjectDetection,
@@ -163,6 +173,46 @@
163173
"instance-segmentation",
164174
"yolov11x-seg",
165175
): YOLOv11InstanceSegmentation,
176+
(
177+
"instance-segmentation",
178+
"yolo26n",
179+
): YOLO26InstanceSegmentation,
180+
(
181+
"instance-segmentation",
182+
"yolo26s",
183+
): YOLO26InstanceSegmentation,
184+
(
185+
"instance-segmentation",
186+
"yolo26m",
187+
): YOLO26InstanceSegmentation,
188+
(
189+
"instance-segmentation",
190+
"yolo26l",
191+
): YOLO26InstanceSegmentation,
192+
(
193+
"instance-segmentation",
194+
"yolo26x",
195+
): YOLO26InstanceSegmentation,
196+
(
197+
"instance-segmentation",
198+
"yolo26n-seg",
199+
): YOLO26InstanceSegmentation,
200+
(
201+
"instance-segmentation",
202+
"yolo26s-seg",
203+
): YOLO26InstanceSegmentation,
204+
(
205+
"instance-segmentation",
206+
"yolo26m-seg",
207+
): YOLO26InstanceSegmentation,
208+
(
209+
"instance-segmentation",
210+
"yolo26l-seg",
211+
): YOLO26InstanceSegmentation,
212+
(
213+
"instance-segmentation",
214+
"yolo26x-seg",
215+
): YOLO26InstanceSegmentation,
166216
("keypoint-detection", "yolov11n"): YOLOv11KeypointsDetection,
167217
("keypoint-detection", "yolov11s"): YOLOv11KeypointsDetection,
168218
("keypoint-detection", "yolov11m"): YOLOv11KeypointsDetection,
@@ -173,6 +223,16 @@
173223
("keypoint-detection", "yolov11m-pose"): YOLOv11KeypointsDetection,
174224
("keypoint-detection", "yolov11l-pose"): YOLOv11KeypointsDetection,
175225
("keypoint-detection", "yolov11x-pose"): YOLOv11KeypointsDetection,
226+
("keypoint-detection", "yolo26n"): YOLO26KeypointsDetection,
227+
("keypoint-detection", "yolo26s"): YOLO26KeypointsDetection,
228+
("keypoint-detection", "yolo26m"): YOLO26KeypointsDetection,
229+
("keypoint-detection", "yolo26l"): YOLO26KeypointsDetection,
230+
("keypoint-detection", "yolo26x"): YOLO26KeypointsDetection,
231+
("keypoint-detection", "yolo26n-pose"): YOLO26KeypointsDetection,
232+
("keypoint-detection", "yolo26s-pose"): YOLO26KeypointsDetection,
233+
("keypoint-detection", "yolo26m-pose"): YOLO26KeypointsDetection,
234+
("keypoint-detection", "yolo26l-pose"): YOLO26KeypointsDetection,
235+
("keypoint-detection", "yolo26x-pose"): YOLO26KeypointsDetection,
176236
("instance-segmentation", "stub"): InstanceSegmentationModelStub,
177237
(
178238
"instance-segmentation",

0 commit comments

Comments
 (0)