Back to Ultralytics

Reference for `ultralytics/nn/tasks.py`

docs/en/reference/nn/tasks.md

8.4.461.6 KB
Original Source

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

<hr>

::: ultralytics.nn.tasks.DetectionModel

<hr>

::: ultralytics.nn.tasks.OBBModel

<hr>

::: ultralytics.nn.tasks.SegmentationModel

<hr>

::: ultralytics.nn.tasks.PoseModel

<hr>

::: ultralytics.nn.tasks.ClassificationModel

<hr>

::: ultralytics.nn.tasks.RTDETRDetectionModel

<hr>

::: ultralytics.nn.tasks.WorldModel

<hr>

::: ultralytics.nn.tasks.YOLOEModel

<hr>

::: ultralytics.nn.tasks.YOLOESegModel

<hr>

::: ultralytics.nn.tasks.Ensemble

<hr>

::: ultralytics.nn.tasks.SafeClass

<hr>

::: ultralytics.nn.tasks.SafeUnpickler

<hr>

::: ultralytics.nn.tasks.temporary_modules

<hr>

::: ultralytics.nn.tasks.torch_safe_load

<hr>

::: ultralytics.nn.tasks.load_checkpoint

<hr>

::: ultralytics.nn.tasks.parse_model

<hr>

::: ultralytics.nn.tasks.yaml_model_load

<hr>

::: ultralytics.nn.tasks.guess_model_scale

<hr>

::: ultralytics.nn.tasks.guess_model_task