File size: 2,010 Bytes
f6228f9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
---

description: Dive into the intricacies of YOLO tasks.py. Learn about DetectionModel, PoseModel and more for powerful AI development.
keywords: 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 🙏!


<br>

## ::: ultralytics.nn.tasks.BaseModel

<br><br><hr><br>

## ::: ultralytics.nn.tasks.DetectionModel

<br><br><hr><br>

## ::: ultralytics.nn.tasks.OBBModel

<br><br><hr><br>

## ::: ultralytics.nn.tasks.SegmentationModel

<br><br><hr><br>

## ::: ultralytics.nn.tasks.PoseModel

<br><br><hr><br>

## ::: ultralytics.nn.tasks.ClassificationModel

<br><br><hr><br>

## ::: ultralytics.nn.tasks.RTDETRDetectionModel

<br><br><hr><br>

## ::: ultralytics.nn.tasks.WorldModel

<br><br><hr><br>

## ::: ultralytics.nn.tasks.Ensemble

<br><br><hr><br>

## ::: ultralytics.nn.tasks.SafeClass

<br><br><hr><br>

## ::: ultralytics.nn.tasks.SafeUnpickler

<br><br><hr><br>

## ::: ultralytics.nn.tasks.temporary_modules



<br><br><hr><br>



## ::: ultralytics.nn.tasks.torch_safe_load



<br><br><hr><br>



## ::: ultralytics.nn.tasks.attempt_load_weights



<br><br><hr><br>



## ::: ultralytics.nn.tasks.attempt_load_one_weight

<br><br><hr><br>

## ::: ultralytics.nn.tasks.parse_model



<br><br><hr><br>



## ::: ultralytics.nn.tasks.yaml_model_load



<br><br><hr><br>



## ::: ultralytics.nn.tasks.guess_model_scale



<br><br><hr><br>



## ::: ultralytics.nn.tasks.guess_model_task



<br><br>