You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

1.5 KiB

description keywords
Explore Ultralytics YOLO docs to understand task-specific models like DetectionModel, PoseModel, RTDETRDetectionModel and more. Plus, learn about ensemble, model loading. Ultralytics, YOLO docs, DetectionModel, SegmentationModel, ClassificationModel, Ensemble, torch_safe_load, yaml_model_load, guess_model_task

BaseModel


::: ultralytics.nn.tasks.BaseModel



DetectionModel


::: ultralytics.nn.tasks.DetectionModel



SegmentationModel


::: ultralytics.nn.tasks.SegmentationModel



PoseModel


::: ultralytics.nn.tasks.PoseModel



ClassificationModel


::: ultralytics.nn.tasks.ClassificationModel



RTDETRDetectionModel


::: ultralytics.nn.tasks.RTDETRDetectionModel



Ensemble


::: ultralytics.nn.tasks.Ensemble



temporary_modules


::: ultralytics.nn.tasks.temporary_modules



torch_safe_load


::: ultralytics.nn.tasks.torch_safe_load



attempt_load_weights


::: ultralytics.nn.tasks.attempt_load_weights



attempt_load_one_weight


::: ultralytics.nn.tasks.attempt_load_one_weight



parse_model


::: ultralytics.nn.tasks.parse_model



yaml_model_load


::: ultralytics.nn.tasks.yaml_model_load



guess_model_scale


::: ultralytics.nn.tasks.guess_model_scale



guess_model_task


::: ultralytics.nn.tasks.guess_model_task