ultralytics/docs/en/reference/nn/tasks.md

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

!!! success "Improvements"

This page is sourced from [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/nn/tasks.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/nn/tasks.py). Have an improvement or example to add? Open a [Pull Request](https://docs.ultralytics.com/help/contributing/) — 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.YOLOEModel

---

::: ultralytics.nn.tasks.YOLOESegModel

---

::: ultralytics.nn.tasks.Ensemble

---

::: ultralytics.nn.tasks.SafeClass

---

::: ultralytics.nn.tasks.SafeUnpickler

---

::: ultralytics.nn.tasks.temporary_modules

---

::: ultralytics.nn.tasks.torch_safe_load

---

::: ultralytics.nn.tasks.load_checkpoint

---

::: ultralytics.nn.tasks.parse_model

---

::: ultralytics.nn.tasks.yaml_model_load

---

::: ultralytics.nn.tasks.guess_model_scale

---

::: ultralytics.nn.tasks.guess_model_task