Skip to content

Commit 054ca0d

Browse files
committed
process yolov12
1 parent 976af37 commit 054ca0d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

roboflow/util/model_processor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,7 @@ def _process_yolo(model_type: str, model_path: str, filename: str) -> str:
136136
class_names.sort(key=lambda x: x[0])
137137
class_names = [x[1] for x in class_names]
138138

139-
if "yolov8" in model_type or "yolov10" in model_type or "yolov11" in model_type:
139+
if "yolov8" in model_type or "yolov10" in model_type or "yolov11" in model_type or "yolov12" in model_type:
140140
# try except for backwards compatibility with older versions of ultralytics
141141
if "-cls" in model_type or model_type.startswith("yolov10") or model_type.startswith("yolov11"):
142142
nc = model_instance.yaml["nc"]

0 commit comments

Comments
 (0)