ultralytics/docs/en/reference/nn/tasks.md
2025-02-25 11:58:34 +08:00

1.6 KiB

description keywords
Dive into the intricacies of YOLO tasks.py. Learn about DetectionModel, PoseModel and more for powerful AI development. Ultralytics, YOLO, nn tasks, DetectionModel, PoseModel, RTDETRDetectionModel, model weights, parse model, AI development

Reference for ultralytics/nn/tasks.py

!!! Note

This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/nn/tasks.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/nn/tasks.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/nn/tasks.py) 🛠️. Thank you 🙏!



::: ultralytics.nn.tasks.BaseModel



::: ultralytics.nn.tasks.DetectionModel



::: ultralytics.nn.tasks.OBBModel



::: ultralytics.nn.tasks.SegmentationModel



::: ultralytics.nn.tasks.PoseModel



::: ultralytics.nn.tasks.ClassificationModel



::: ultralytics.nn.tasks.RTDETRDetectionModel



::: ultralytics.nn.tasks.WorldModel



::: ultralytics.nn.tasks.Ensemble



::: ultralytics.nn.tasks.temporary_modules



::: ultralytics.nn.tasks.torch_safe_load



::: ultralytics.nn.tasks.attempt_load_weights



::: ultralytics.nn.tasks.attempt_load_one_weight



::: ultralytics.nn.tasks.parse_model



::: ultralytics.nn.tasks.yaml_model_load



::: ultralytics.nn.tasks.guess_model_scale



::: ultralytics.nn.tasks.guess_model_task