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.8 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