Π‘ΡΡΠ»ΠΊΠ° Π΄Π»Ρ ultralytics/nn/tasks.py
ΠΡΠΈΠΌΠ΅ΡΠ°Π½ΠΈΠ΅
ΠΡΠΎΡ ΡΠ°ΠΉΠ» Π΄ΠΎΡΡΡΠΏΠ΅Π½ ΠΏΠΎ Π°Π΄ΡΠ΅ΡΡ https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/nn/tasks .py. ΠΡΠ»ΠΈ ΡΡ Π·Π°ΠΌΠ΅ΡΠΈΠ» ΠΏΡΠΎΠ±Π»Π΅ΠΌΡ, ΠΏΠΎΠΆΠ°Π»ΡΠΉΡΡΠ°, ΠΏΠΎΠΌΠΎΠ³ΠΈ ΠΈΡΠΏΡΠ°Π²ΠΈΡΡ Π΅Π΅, ΡΠΎΠ·Π΄Π°Π² Pull Request π οΈ. Π‘ΠΏΠ°ΡΠΈΠ±ΠΎ π!
ultralytics.nn.tasks.BaseModel
ΠΠ°Π·Ρ: Module
ΠΠ»Π°ΡΡ BaseModel ΡΠ»ΡΠΆΠΈΡ Π±Π°Π·ΠΎΠ²ΡΠΌ ΠΊΠ»Π°ΡΡΠΎΠΌ Π΄Π»Ρ Π²ΡΠ΅Ρ ΠΌΠΎΠ΄Π΅Π»Π΅ΠΉ ΡΠ΅ΠΌΠ΅ΠΉΡΡΠ²Π° Ultralytics YOLO .
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 |
|
forward(x, *args, **kwargs)
ΠΡΠΎΡ
ΠΎΠΆΠ΄Π΅Π½ΠΈΠ΅ ΠΌΠΎΠ΄Π΅Π»ΠΈ Π²ΠΏΠ΅ΡΠ΅Π΄ Π½Π° ΠΎΠ΄Π½ΠΎΠΌ ΠΌΠ°ΡΡΡΠ°Π±Π΅. ΠΠ±Π΅ΡΡΠΊΠ° Π΄Π»Ρ _forward_once
ΠΌΠ΅ΡΠΎΠ΄.
ΠΠ°ΡΠ°ΠΌΠ΅ΡΡΡ:
ΠΠΌΡ | Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ | ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ |
---|---|---|---|
x |
Tensor | dict
|
ΠΡ ΠΎΠ΄Π½ΠΎΠ΅ ΠΈΠ·ΠΎΠ±ΡΠ°ΠΆΠ΅Π½ΠΈΠ΅ tensor ΠΈΠ»ΠΈ Π΄ΠΈΠΊΡΠ°Π½Ρ, Π²ΠΊΠ»ΡΡΠ°ΡΡΠΈΠΉ ΠΈΠ·ΠΎΠ±ΡΠ°ΠΆΠ΅Π½ΠΈΠ΅ tensor ΠΈ ΠΌΠ΅ΡΠΊΠΈ gt. |
ΡΡΠ΅Π±ΡΠ΅ΡΡΡ |
ΠΠΎΠ·Π²ΡΠ°ΡΠ°Π΅ΡΡΡ:
Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ |
---|---|
Tensor
|
ΠΡΡ ΠΎΠ΄Π½ΠΎΠΉ ΡΠΈΠ³Π½Π°Π» ΡΠ΅ΡΠΈ. |
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
fuse(verbose=True)
ΠΠΎΠ΄ΡΠΎΠ΅Π΄ΠΈΠ½ΠΈ Conv2d()
ΠΈ BatchNorm2d()
ΡΠ»ΠΎΠΈ ΠΌΠΎΠ΄Π΅Π»ΠΈ Π² ΠΎΠ΄ΠΈΠ½ ΡΠ»ΠΎΠΉ, ΡΡΠΎΠ±Ρ ΠΏΠΎΠ²ΡΡΠΈΡΡ
ΡΡΡΠ΅ΠΊΡΠΈΠ²Π½ΠΎΡΡΡ Π²ΡΡΠΈΡΠ»Π΅Π½ΠΈΠΉ.
ΠΠΎΠ·Π²ΡΠ°ΡΠ°Π΅ΡΡΡ:
Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ |
---|---|
Module
|
ΠΠΎΠ·Π²ΡΠ°ΡΠ°Π΅ΡΡΡ ΡΠ»ΠΈΡΠ°Ρ ΠΌΠΎΠ΄Π΅Π»Ρ. |
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
info(detailed=False, verbose=True, imgsz=640)
ΠΠ΅ΡΠ°ΡΠ°Π΅Ρ ΠΈΠ½ΡΠΎΡΠΌΠ°ΡΠΈΡ ΠΎ ΠΌΠΎΠ΄Π΅Π»ΠΈ.
ΠΠ°ΡΠ°ΠΌΠ΅ΡΡΡ:
ΠΠΌΡ | Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ | ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ |
---|---|---|---|
detailed |
bool
|
ΠΡΠ»ΠΈ True, ΡΠΎ Π²ΡΠ²ΠΎΠ΄ΠΈΡ ΠΏΠΎΠ΄ΡΠΎΠ±Π½ΡΡ ΠΈΠ½ΡΠΎΡΠΌΠ°ΡΠΈΡ ΠΎ ΠΌΠΎΠ΄Π΅Π»ΠΈ. ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ ΡΡΡΠ°Π½ΠΎΠ²Π»Π΅Π½ΠΎ Π·Π½Π°ΡΠ΅Π½ΠΈΠ΅ False |
False
|
verbose |
bool
|
ΠΡΠ»ΠΈ True, Π²ΡΠ²ΠΎΠ΄ΠΈΡ ΠΈΠ½ΡΠΎΡΠΌΠ°ΡΠΈΡ ΠΎ ΠΌΠΎΠ΄Π΅Π»ΠΈ. ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ ΡΡΡΠ°Π½ΠΎΠ²Π»Π΅Π½ΠΎ Π·Π½Π°ΡΠ΅Π½ΠΈΠ΅ False |
True
|
imgsz |
int
|
ΡΠ°Π·ΠΌΠ΅Ρ ΠΈΠ·ΠΎΠ±ΡΠ°ΠΆΠ΅Π½ΠΈΡ, Π½Π° ΠΊΠΎΡΠΎΡΠΎΠΌ Π±ΡΠ΄Π΅Ρ ΠΎΠ±ΡΡΠ°ΡΡΡΡ ΠΌΠΎΠ΄Π΅Π»Ρ. ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ 640 |
640
|
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
init_criterion()
is_fused(thresh=10)
ΠΡΠΎΠ²Π΅ΡΡ, Π½Π΅ ΠΈΠΌΠ΅Π΅Ρ Π»ΠΈ ΠΌΠΎΠ΄Π΅Π»Ρ ΠΌΠ΅Π½ΡΡΠ΅ ΠΎΠΏΡΠ΅Π΄Π΅Π»Π΅Π½Π½ΠΎΠ³ΠΎ ΠΏΠΎΡΠΎΠ³Π° ΡΠ»ΠΎΠ΅Π² BatchNorm.
ΠΠ°ΡΠ°ΠΌΠ΅ΡΡΡ:
ΠΠΌΡ | Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ | ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ |
---|---|---|---|
thresh |
int
|
ΠΠΎΡΠΎΠ³ΠΎΠ²ΠΎΠ΅ ΠΊΠΎΠ»ΠΈΡΠ΅ΡΡΠ²ΠΎ ΡΠ»ΠΎΠ΅Π² BatchNorm. ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ ΡΡΠΎ 10. |
10
|
ΠΠΎΠ·Π²ΡΠ°ΡΠ°Π΅ΡΡΡ:
Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ |
---|---|
bool
|
True, Π΅ΡΠ»ΠΈ ΠΊΠΎΠ»ΠΈΡΠ΅ΡΡΠ²ΠΎ ΡΠ»ΠΎΠ΅Π² BatchNorm Π² ΠΌΠΎΠ΄Π΅Π»ΠΈ ΠΌΠ΅Π½ΡΡΠ΅ ΠΏΠΎΡΠΎΠ³Π°, False - Π² ΠΏΡΠΎΡΠΈΠ²Π½ΠΎΠΌ ΡΠ»ΡΡΠ°Π΅. |
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
load(weights, verbose=True)
ΠΠ°Π³ΡΡΠ·ΠΈ Π²Π΅ΡΠΎΠ²ΡΠ΅ ΠΊΠΎΡΡΡΠΈΡΠΈΠ΅Π½ΡΡ Π² ΠΌΠΎΠ΄Π΅Π»Ρ.
ΠΠ°ΡΠ°ΠΌΠ΅ΡΡΡ:
ΠΠΌΡ | Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ | ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ |
---|---|---|---|
weights |
dict | Module
|
ΠΡΠ΅Π΄Π²Π°ΡΠΈΡΠ΅Π»ΡΠ½ΠΎ ΠΎΠ±ΡΡΠ΅Π½Π½ΡΠ΅ Π²Π΅ΡΠ°, ΠΊΠΎΡΠΎΡΡΠ΅ Π½ΡΠΆΠ½ΠΎ Π·Π°Π³ΡΡΠ·ΠΈΡΡ. |
ΡΡΠ΅Π±ΡΠ΅ΡΡΡ |
verbose |
bool
|
ΠΡΠΆΠ½ΠΎ Π»ΠΈ Π·Π°ΠΏΠΈΡΡΠ²Π°ΡΡ Π² ΠΆΡΡΠ½Π°Π» ΠΏΡΠΎΠ³ΡΠ΅ΡΡ ΠΏΠ΅ΡΠ΅Π½ΠΎΡΠ°. ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ ΡΡΡΠ°Π½ΠΎΠ²Π»Π΅Π½ΠΎ Π·Π½Π°ΡΠ΅Π½ΠΈΠ΅ True. |
True
|
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
loss(batch, preds=None)
ΠΡΡΠΈΡΠ»ΠΈ ΠΏΠΎΡΠ΅ΡΡ.
ΠΠ°ΡΠ°ΠΌΠ΅ΡΡΡ:
ΠΠΌΡ | Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ | ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ |
---|---|---|---|
batch |
dict
|
ΠΠ°ΡΡΠΈΡ Π΄Π»Ρ Π²ΡΡΠΈΡΠ»Π΅Π½ΠΈΡ ΠΏΠΎΡΠ΅ΡΡ |
ΡΡΠ΅Π±ΡΠ΅ΡΡΡ |
preds |
Tensor | List[Tensor]
|
ΠΡΠΎΠ³Π½ΠΎΠ·Ρ. |
None
|
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
predict(x, profile=False, visualize=False, augment=False, embed=None)
ΠΡΠΏΠΎΠ»Π½ΠΈ ΠΏΡΡΠΌΠΎΠΉ ΠΏΡΠΎΡ ΠΎΠ΄ ΠΏΠΎ ΡΠ΅ΡΠΈ.
ΠΠ°ΡΠ°ΠΌΠ΅ΡΡΡ:
ΠΠΌΡ | Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ | ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ |
---|---|---|---|
x |
Tensor
|
ΠΡ ΠΎΠ΄Π½ΡΠ΅ Π΄Π°Π½Π½ΡΠ΅ tensor Π΄Π»Ρ ΠΌΠΎΠ΄Π΅Π»ΠΈ. |
ΡΡΠ΅Π±ΡΠ΅ΡΡΡ |
profile |
bool
|
ΠΡΠ²Π΅Π΄ΠΈ Π²ΡΠ΅ΠΌΡ Π²ΡΡΠΈΡΠ»Π΅Π½ΠΈΡ ΠΊΠ°ΠΆΠ΄ΠΎΠ³ΠΎ ΡΠ»ΠΎΡ, Π΅ΡΠ»ΠΈ Π·Π½Π°ΡΠ΅Π½ΠΈΠ΅ True, ΠΏΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ False. |
False
|
visualize |
bool
|
Π‘ΠΎΡ ΡΠ°Π½ΡΠΉ ΠΊΠ°ΡΡΡ Ρ Π°ΡΠ°ΠΊΡΠ΅ΡΠΈΡΡΠΈΠΊ ΠΌΠΎΠ΄Π΅Π»ΠΈ, Π΅ΡΠ»ΠΈ Π·Π½Π°ΡΠ΅Π½ΠΈΠ΅ True, ΠΏΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ False. |
False
|
augment |
bool
|
ΠΠΎΠΏΠΎΠ»Π½ΠΈ ΠΈΠ·ΠΎΠ±ΡΠ°ΠΆΠ΅Π½ΠΈΠ΅ Π²ΠΎ Π²ΡΠ΅ΠΌΡ ΠΏΡΠ΅Π΄ΡΠΊΠ°Π·Π°Π½ΠΈΡ, ΠΏΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ False. |
False
|
embed |
list
|
Π‘ΠΏΠΈΡΠΎΠΊ Π²Π΅ΠΊΡΠΎΡΠΎΠ² ΠΏΡΠΈΠ·Π½Π°ΠΊΠΎΠ²/Π²ΠΊΡΠ°ΠΏΠ»Π΅Π½ΠΈΠΉ, ΠΊΠΎΡΠΎΡΡΠ΅ Π½ΡΠΆΠ½ΠΎ Π²Π΅ΡΠ½ΡΡΡ. |
None
|
ΠΠΎΠ·Π²ΡΠ°ΡΠ°Π΅ΡΡΡ:
Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ |
---|---|
Tensor
|
ΠΠΎΡΠ»Π΅Π΄Π½ΠΈΠΉ Π²ΡΡ ΠΎΠ΄ ΠΌΠΎΠ΄Π΅Π»ΠΈ. |
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
ultralytics.nn.tasks.DetectionModel
ΠΠ°Π·Ρ: BaseModel
YOLOv8 ΠΌΠΎΠ΄Π΅Π»Ρ ΠΎΠ±Π½Π°ΡΡΠΆΠ΅Π½ΠΈΡ.
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
__init__(cfg='yolov8n.yaml', ch=3, nc=None, verbose=True)
ΠΠ½ΠΈΡΠΈΠ°Π»ΠΈΠ·ΠΈΡΡΠΉ ΠΌΠΎΠ΄Π΅Π»Ρ ΠΎΠ±Π½Π°ΡΡΠΆΠ΅Π½ΠΈΡ YOLOv8 Ρ Π·Π°Π΄Π°Π½Π½ΡΠΌ ΠΊΠΎΠ½ΡΠΈΠ³ΠΎΠΌ ΠΈ ΠΏΠ°ΡΠ°ΠΌΠ΅ΡΡΠ°ΠΌΠΈ.
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
ultralytics.nn.tasks.OBBModel
ΠΠ°Π·Ρ: DetectionModel
YOLOv8 ΠΠΎΠ΄Π΅Π»Ρ Oriented Bounding Box (OBB).
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
__init__(cfg='yolov8n-obb.yaml', ch=3, nc=None, verbose=True)
ΠΠ½ΠΈΡΠΈΠ°Π»ΠΈΠ·ΠΈΡΡΠΉ ΠΌΠΎΠ΄Π΅Π»Ρ YOLOv8 OBB Ρ Π·Π°Π΄Π°Π½Π½ΡΠΌ ΠΊΠΎΠ½ΡΠΈΠ³ΠΎΠΌ ΠΈ ΠΏΠ°ΡΠ°ΠΌΠ΅ΡΡΠ°ΠΌΠΈ.
ultralytics.nn.tasks.SegmentationModel
ΠΠ°Π·Ρ: DetectionModel
YOLOv8 ΠΠΎΠ΄Π΅Π»Ρ ΡΠ΅Π³ΠΌΠ΅Π½ΡΠ°ΡΠΈΠΈ.
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
__init__(cfg='yolov8n-seg.yaml', ch=3, nc=None, verbose=True)
ΠΠ½ΠΈΡΠΈΠ°Π»ΠΈΠ·ΠΈΡΡΠΉ ΠΌΠΎΠ΄Π΅Π»Ρ ΡΠ΅Π³ΠΌΠ΅Π½ΡΠ°ΡΠΈΠΈ YOLOv8 Ρ Π·Π°Π΄Π°Π½Π½ΡΠΌ ΠΊΠΎΠ½ΡΠΈΠ³ΠΎΠΌ ΠΈ ΠΏΠ°ΡΠ°ΠΌΠ΅ΡΡΠ°ΠΌΠΈ.
ultralytics.nn.tasks.PoseModel
ΠΠ°Π·Ρ: DetectionModel
YOLOv8 ΠΠΎΠ΄Π΅Π»Ρ ΠΏΠΎΠ·Ρ.
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
__init__(cfg='yolov8n-pose.yaml', ch=3, nc=None, data_kpt_shape=(None, None), verbose=True)
ΠΠ½ΠΈΡΠΈΠ°Π»ΠΈΠ·ΠΈΡΡΠΉ ΠΌΠΎΠ΄Π΅Π»Ρ YOLOv8 Pose.
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
ultralytics.nn.tasks.ClassificationModel
ΠΠ°Π·Ρ: BaseModel
YOLOv8 ΠΌΠΎΠ΄Π΅Π»Ρ ΠΊΠ»Π°ΡΡΠΈΡΠΈΠΊΠ°ΡΠΈΠΈ.
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
__init__(cfg='yolov8n-cls.yaml', ch=3, nc=None, verbose=True)
ΠΠ½ΠΈΡΠΈΠΈΡΡΠΉ ClassificationModel Ρ YAML, ΠΊΠ°Π½Π°Π»Π°ΠΌΠΈ, ΠΊΠΎΠ»ΠΈΡΠ΅ΡΡΠ²ΠΎΠΌ ΠΊΠ»Π°ΡΡΠΎΠ², ΡΠ»Π°Π³ΠΎΠΌ verbose.
init_criterion()
reshape_outputs(model, nc)
staticmethod
ΠΠ±Π½ΠΎΠ²ΠΈ ΠΌΠΎΠ΄Π΅Π»Ρ ΠΊΠ»Π°ΡΡΠΈΡΠΈΠΊΠ°ΡΠΈΠΈ TorchVision Π½Π° ΠΊΠ»Π°ΡΡ 'n', Π΅ΡΠ»ΠΈ ΡΡΠΎ Π½Π΅ΠΎΠ±Ρ ΠΎΠ΄ΠΈΠΌΠΎ.
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
ultralytics.nn.tasks.RTDETRDetectionModel
ΠΠ°Π·Ρ: DetectionModel
ΠΠ»Π°ΡΡ ΠΌΠΎΠ΄Π΅Π»Π΅ΠΉ ΠΎΠ±Π½Π°ΡΡΠΆΠ΅Π½ΠΈΡ RTDETR (Real-time DEtection and Tracking using Transformers).
ΠΡΠΎΡ ΠΊΠ»Π°ΡΡ ΠΎΡΠ²Π΅ΡΠ°Π΅Ρ Π·Π° ΠΏΠΎΡΡΡΠΎΠ΅Π½ΠΈΠ΅ Π°ΡΡ ΠΈΡΠ΅ΠΊΡΡΡΡ RTDETR, ΠΎΠΏΡΠ΅Π΄Π΅Π»Π΅Π½ΠΈΠ΅ ΡΡΠ½ΠΊΡΠΈΠΉ ΠΏΠΎΡΠ΅ΡΡ ΠΈ ΠΎΠ±Π»Π΅Π³ΡΠ΅Π½ΠΈΠ΅ ΠΏΡΠΎΡΠ΅ΡΡΠΎΠ² ΠΎΠ±ΡΡΠ΅Π½ΠΈΡ ΠΈ Π²ΡΠ²ΠΎΠ΄Π°. ΠΏΡΠΎΡΠ΅ΡΡΡ ΠΎΠ±ΡΡΠ΅Π½ΠΈΡ ΠΈ Π²ΡΠ²ΠΎΠ΄Π°. RTDETR - ΡΡΠΎ ΠΌΠΎΠ΄Π΅Π»Ρ ΠΎΠ±Π½Π°ΡΡΠΆΠ΅Π½ΠΈΡ ΠΈ ΠΎΡΡΠ»Π΅ΠΆΠΈΠ²Π°Π½ΠΈΡ ΠΎΠ±ΡΠ΅ΠΊΡΠΎΠ², ΠΊΠΎΡΠΎΡΠ°Ρ ΡΠ°ΡΡΠΈΡΡΠ΅ΡΡΡ ΠΎΡ Π±Π°Π·ΠΎΠ²ΠΎΠ³ΠΎ ΠΊΠ»Π°ΡΡΠ° DetectionModel Π±Π°Π·ΠΎΠ²ΠΎΠ³ΠΎ ΠΊΠ»Π°ΡΡΠ°.
ΠΡΡΠΈΠ±ΡΡΡ:
ΠΠΌΡ | Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ |
---|---|---|
cfg |
str
|
ΠΡΡΡ ΠΊ ΡΠ°ΠΉΠ»Ρ ΠΊΠΎΠ½ΡΠΈΠ³ΡΡΠ°ΡΠΈΠΈ ΠΈΠ»ΠΈ ΡΡΡΠΎΠΊΠ° ΠΏΡΠ΅Π΄ΡΡΡΠ°Π½ΠΎΠ²ΠΊΠΈ. ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ ΡΡΠΎ 'rtdetr-l.yaml'. |
ch |
int
|
ΠΠΎΠ»ΠΈΡΠ΅ΡΡΠ²ΠΎ Π²Ρ ΠΎΠ΄Π½ΡΡ ΠΊΠ°Π½Π°Π»ΠΎΠ². ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ - 3 (RGB). |
nc |
int
|
ΠΠΎΠ»ΠΈΡΠ΅ΡΡΠ²ΠΎ ΠΊΠ»Π°ΡΡΠΎΠ² Π΄Π»Ρ ΠΎΠ±Π½Π°ΡΡΠΆΠ΅Π½ΠΈΡ ΠΎΠ±ΡΠ΅ΠΊΡΠΎΠ². ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ - None. |
verbose |
bool
|
Π£ΠΊΠ°Π·ΡΠ²Π°Π΅Ρ, ΠΏΠΎΠΊΠ°Π·ΡΠ²Π°ΡΡ Π»ΠΈ ΡΠ²ΠΎΠ΄Π½ΡΡ ΡΡΠ°ΡΠΈΡΡΠΈΠΊΡ Π²ΠΎ Π²ΡΠ΅ΠΌΡ ΠΈΠ½ΠΈΡΠΈΠ°Π»ΠΈΠ·Π°ΡΠΈΠΈ. ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ ΡΡΠΎ True. |
ΠΠ΅ΡΠΎΠ΄Ρ:
ΠΠΌΡ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ |
---|---|
init_criterion |
ΠΠ½ΠΈΡΠΈΠ°Π»ΠΈΠ·ΠΈΡΡΠ΅Ρ ΠΊΡΠΈΡΠ΅ΡΠΈΠΉ, ΠΈΡΠΏΠΎΠ»ΡΠ·ΡΠ΅ΠΌΡΠΉ Π΄Π»Ρ ΡΠ°ΡΡΠ΅ΡΠ° ΠΏΠΎΡΠ΅ΡΡ. |
loss |
ΠΡΡΠΈΡΠ»ΡΠ΅Ρ ΠΈ Π²ΠΎΠ·Π²ΡΠ°ΡΠ°Π΅Ρ ΠΏΠΎΡΠ΅ΡΠΈ Π²ΠΎ Π²ΡΠ΅ΠΌΡ ΡΡΠ΅Π½ΠΈΡΠΎΠ²ΠΊΠΈ. |
predict |
ΠΡΠΏΠΎΠ»Π½ΡΠ΅Ρ ΠΏΡΡΠΌΠΎΠΉ ΠΏΡΠΎΡ ΠΎΠ΄ ΠΏΠΎ ΡΠ΅ΡΠΈ ΠΈ Π²ΠΎΠ·Π²ΡΠ°ΡΠ°Π΅Ρ Π²ΡΡ ΠΎΠ΄. |
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 |
|
__init__(cfg='rtdetr-l.yaml', ch=3, nc=None, verbose=True)
ΠΠ½ΠΈΡΠΈΠ°Π»ΠΈΠ·ΠΈΡΡΠΉ ΠΌΠΎΠ΄Π΅Π»Ρ RTDETRDetectionModel.
ΠΠ°ΡΠ°ΠΌΠ΅ΡΡΡ:
ΠΠΌΡ | Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ | ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ |
---|---|---|---|
cfg |
str
|
ΠΠΌΡ ΡΠ°ΠΉΠ»Π° ΠΊΠΎΠ½ΡΠΈΠ³ΡΡΠ°ΡΠΈΠΈ ΠΈΠ»ΠΈ ΠΏΡΡΡ ΠΊ Π½Π΅ΠΌΡ. |
'rtdetr-l.yaml'
|
ch |
int
|
ΠΠΎΠ»ΠΈΡΠ΅ΡΡΠ²ΠΎ Π²Ρ ΠΎΠ΄Π½ΡΡ ΠΊΠ°Π½Π°Π»ΠΎΠ². |
3
|
nc |
int
|
ΠΠΎΠ»ΠΈΡΠ΅ΡΡΠ²ΠΎ ΠΊΠ»Π°ΡΡΠΎΠ². ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ - None. |
None
|
verbose |
bool
|
ΠΡΠ²ΠΎΠ΄ΠΈ Π΄ΠΎΠΏΠΎΠ»Π½ΠΈΡΠ΅Π»ΡΠ½ΡΡ ΠΈΠ½ΡΠΎΡΠΌΠ°ΡΠΈΡ Π²ΠΎ Π²ΡΠ΅ΠΌΡ ΠΈΠ½ΠΈΡΠΈΠ°Π»ΠΈΠ·Π°ΡΠΈΠΈ. ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ ΡΡΡΠ°Π½ΠΎΠ²Π»Π΅Π½ΠΎ Π·Π½Π°ΡΠ΅Π½ΠΈΠ΅ True. |
True
|
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
init_criterion()
ΠΠ½ΠΈΡΠΈΠ°Π»ΠΈΠ·ΠΈΡΡΠΉ ΠΊΡΠΈΡΠ΅ΡΠΈΠΉ ΠΏΠΎΡΠ΅ΡΡ Π΄Π»Ρ ΠΌΠΎΠ΄Π΅Π»ΠΈ RTDETRDetectionModel.
loss(batch, preds=None)
ΠΡΡΠΈΡΠ»ΠΈ ΠΏΠΎΡΠ΅ΡΠΈ Π΄Π»Ρ Π·Π°Π΄Π°Π½Π½ΠΎΠΉ ΠΏΠ°ΡΡΠΈΠΈ Π΄Π°Π½Π½ΡΡ .
ΠΠ°ΡΠ°ΠΌΠ΅ΡΡΡ:
ΠΠΌΡ | Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ | ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ |
---|---|---|---|
batch |
dict
|
Π‘Π»ΠΎΠ²Π°ΡΡ, ΡΠΎΠ΄Π΅ΡΠΆΠ°ΡΠΈΠΉ Π΄Π°Π½Π½ΡΠ΅ ΠΎΠ± ΠΈΠ·ΠΎΠ±ΡΠ°ΠΆΠ΅Π½ΠΈΡΡ ΠΈ ΠΌΠ΅ΡΠΊΠ°Ρ . |
ΡΡΠ΅Π±ΡΠ΅ΡΡΡ |
preds |
Tensor
|
ΠΡΠ΅Π΄Π²Π°ΡΠΈΡΠ΅Π»ΡΠ½ΠΎ Π²ΡΡΠΈΡΠ»Π΅Π½Π½ΡΠ΅ ΠΏΡΠ΅Π΄ΡΠΊΠ°Π·Π°Π½ΠΈΡ ΠΌΠΎΠ΄Π΅Π»ΠΈ. ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ ΡΡΡΠ°Π½ΠΎΠ²Π»Π΅Π½ΠΎ Π·Π½Π°ΡΠ΅Π½ΠΈΠ΅ "ΠΠ΅Ρ". |
None
|
ΠΠΎΠ·Π²ΡΠ°ΡΠ°Π΅ΡΡΡ:
Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ |
---|---|
tuple
|
ΠΠΎΡΡΠ΅ΠΆ, ΡΠΎΠ΄Π΅ΡΠΆΠ°ΡΠΈΠΉ ΠΎΠ±ΡΠΈΠΉ ΠΏΡΠΎΠΈΠ³ΡΡΡ ΠΈ ΠΎΡΠ½ΠΎΠ²Π½ΡΠ΅ ΡΡΠΈ ΠΏΡΠΎΠΈΠ³ΡΡΡΠ° Π² tensor. |
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
predict(x, profile=False, visualize=False, batch=None, augment=False, embed=None)
ΠΡΠΏΠΎΠ»Π½ΠΈ ΠΏΡΡΠΌΠΎΠ΅ ΠΏΡΠΎΡ ΠΎΠΆΠ΄Π΅Π½ΠΈΠ΅ ΡΠ΅ΡΠ΅Π· ΠΌΠΎΠ΄Π΅Π»Ρ.
ΠΠ°ΡΠ°ΠΌΠ΅ΡΡΡ:
ΠΠΌΡ | Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ | ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ |
---|---|---|---|
x |
Tensor
|
ΠΡ ΠΎΠ΄ tensor. |
ΡΡΠ΅Π±ΡΠ΅ΡΡΡ |
profile |
bool
|
ΠΡΠ»ΠΈ True, ΡΠΎ ΠΏΡΠΎΡΠΈΠ»ΠΈΡΡΠΉ Π²ΡΠ΅ΠΌΡ Π²ΡΡΠΈΡΠ»Π΅Π½ΠΈΠΉ Π΄Π»Ρ ΠΊΠ°ΠΆΠ΄ΠΎΠ³ΠΎ ΡΠ»ΠΎΡ. ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ ΡΡΡΠ°Π½ΠΎΠ²Π»Π΅Π½ΠΎ Π·Π½Π°ΡΠ΅Π½ΠΈΠ΅ False. |
False
|
visualize |
bool
|
ΠΡΠ»ΠΈ True, ΡΠΎΡ ΡΠ°Π½ΡΠΉ ΠΊΠ°ΡΡΡ Ρ Π°ΡΠ°ΠΊΡΠ΅ΡΠΈΡΡΠΈΠΊ Π΄Π»Ρ Π²ΠΈΠ·ΡΠ°Π»ΠΈΠ·Π°ΡΠΈΠΈ. ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ ΡΡΡΠ°Π½ΠΎΠ²Π»Π΅Π½ΠΎ Π·Π½Π°ΡΠ΅Π½ΠΈΠ΅ False. |
False
|
batch |
dict
|
ΠΡΡΠΈΠ½Π½ΡΠ΅ Π΄Π°Π½Π½ΡΠ΅ Π΄Π»Ρ ΠΎΡΠ΅Π½ΠΊΠΈ. ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ ΡΡΡΠ°Π½ΠΎΠ²Π»Π΅Π½ΠΎ Π·Π½Π°ΡΠ΅Π½ΠΈΠ΅ "ΠΠ΅Ρ". |
None
|
augment |
bool
|
ΠΡΠ»ΠΈ Π·Π½Π°ΡΠ΅Π½ΠΈΠ΅ ΡΠ°Π²Π½ΠΎ True, ΡΠΎ Π²ΠΎ Π²ΡΠ΅ΠΌΡ Π²ΡΠ²ΠΎΠ΄Π° Π±ΡΠ΄Π΅Ρ Π²ΡΠΏΠΎΠ»Π½ΡΡΡΡΡ ΡΠ°ΡΡΠΈΡΠ΅Π½ΠΈΠ΅ Π΄Π°Π½Π½ΡΡ . ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ ΡΡΡΠ°Π½ΠΎΠ²Π»Π΅Π½ΠΎ Π·Π½Π°ΡΠ΅Π½ΠΈΠ΅ False. |
False
|
embed |
list
|
Π‘ΠΏΠΈΡΠΎΠΊ Π²Π΅ΠΊΡΠΎΡΠΎΠ² ΠΏΡΠΈΠ·Π½Π°ΠΊΠΎΠ²/Π²ΠΊΡΠ°ΠΏΠ»Π΅Π½ΠΈΠΉ, ΠΊΠΎΡΠΎΡΡΠ΅ Π½ΡΠΆΠ½ΠΎ Π²Π΅ΡΠ½ΡΡΡ. |
None
|
ΠΠΎΠ·Π²ΡΠ°ΡΠ°Π΅ΡΡΡ:
Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ |
---|---|
Tensor
|
ΠΡΡ ΠΎΠ΄ ΠΌΠΎΠ΄Π΅Π»ΠΈ tensor. |
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
ultralytics.nn.tasks.WorldModel
ΠΠ°Π·Ρ: DetectionModel
YOLOv8 ΠΠΈΡΠΎΠ²Π°Ρ ΠΌΠΎΠ΄Π΅Π»Ρ.
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 |
|
__init__(cfg='yolov8s-world.yaml', ch=3, nc=None, verbose=True)
ΠΠ½ΠΈΡΠΈΠ°Π»ΠΈΠ·ΠΈΡΡΠΉ ΠΌΠΎΠ΄Π΅Π»Ρ ΠΌΠΈΡΠ° YOLOv8 Ρ Π·Π°Π΄Π°Π½Π½ΡΠΌ ΠΊΠΎΠ½ΡΠΈΠ³ΠΎΠΌ ΠΈ ΠΏΠ°ΡΠ°ΠΌΠ΅ΡΡΠ°ΠΌΠΈ.
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
loss(batch, preds=None)
ΠΡΡΠΈΡΠ»ΠΈ ΠΏΠΎΡΠ΅ΡΡ.
ΠΠ°ΡΠ°ΠΌΠ΅ΡΡΡ:
ΠΠΌΡ | Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ | ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ |
---|---|---|---|
batch |
dict
|
ΠΠ°ΠΊΠ΅Ρ Π΄Π»Ρ Π²ΡΡΠΈΡΠ»Π΅Π½ΠΈΡ ΠΏΠΎΡΠ΅ΡΡ. |
ΡΡΠ΅Π±ΡΠ΅ΡΡΡ |
preds |
Tensor | List[Tensor]
|
ΠΡΠΎΠ³Π½ΠΎΠ·Ρ. |
None
|
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
predict(x, profile=False, visualize=False, txt_feats=None, augment=False, embed=None)
ΠΡΠΏΠΎΠ»Π½ΠΈ ΠΏΡΡΠΌΠΎΠ΅ ΠΏΡΠΎΡ ΠΎΠΆΠ΄Π΅Π½ΠΈΠ΅ ΡΠ΅ΡΠ΅Π· ΠΌΠΎΠ΄Π΅Π»Ρ.
ΠΠ°ΡΠ°ΠΌΠ΅ΡΡΡ:
ΠΠΌΡ | Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ | ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ |
---|---|---|---|
x |
Tensor
|
ΠΡ ΠΎΠ΄ tensor. |
ΡΡΠ΅Π±ΡΠ΅ΡΡΡ |
profile |
bool
|
ΠΡΠ»ΠΈ True, ΡΠΎ ΠΏΡΠΎΡΠΈΠ»ΠΈΡΡΠΉ Π²ΡΠ΅ΠΌΡ Π²ΡΡΠΈΡΠ»Π΅Π½ΠΈΠΉ Π΄Π»Ρ ΠΊΠ°ΠΆΠ΄ΠΎΠ³ΠΎ ΡΠ»ΠΎΡ. ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ ΡΡΡΠ°Π½ΠΎΠ²Π»Π΅Π½ΠΎ Π·Π½Π°ΡΠ΅Π½ΠΈΠ΅ False. |
False
|
visualize |
bool
|
ΠΡΠ»ΠΈ True, ΡΠΎΡ ΡΠ°Π½ΡΠΉ ΠΊΠ°ΡΡΡ Ρ Π°ΡΠ°ΠΊΡΠ΅ΡΠΈΡΡΠΈΠΊ Π΄Π»Ρ Π²ΠΈΠ·ΡΠ°Π»ΠΈΠ·Π°ΡΠΈΠΈ. ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ ΡΡΡΠ°Π½ΠΎΠ²Π»Π΅Π½ΠΎ Π·Π½Π°ΡΠ΅Π½ΠΈΠ΅ False. |
False
|
txt_feats |
Tensor
|
Π₯Π°ΡΠ°ΠΊΡΠ΅ΡΠΈΡΡΠΈΠΊΠΈ ΡΠ΅ΠΊΡΡΠ°, ΠΈΡΠΏΠΎΠ»ΡΠ·ΡΠΉ ΠΈΡ , Π΅ΡΠ»ΠΈ ΠΎΠ½ΠΈ Π·Π°Π΄Π°Π½Ρ. ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ - None. |
None
|
augment |
bool
|
ΠΡΠ»ΠΈ Π·Π½Π°ΡΠ΅Π½ΠΈΠ΅ ΡΠ°Π²Π½ΠΎ True, ΡΠΎ Π²ΠΎ Π²ΡΠ΅ΠΌΡ Π²ΡΠ²ΠΎΠ΄Π° Π±ΡΠ΄Π΅Ρ Π²ΡΠΏΠΎΠ»Π½ΡΡΡΡΡ ΡΠ°ΡΡΠΈΡΠ΅Π½ΠΈΠ΅ Π΄Π°Π½Π½ΡΡ . ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ ΡΡΡΠ°Π½ΠΎΠ²Π»Π΅Π½ΠΎ Π·Π½Π°ΡΠ΅Π½ΠΈΠ΅ False. |
False
|
embed |
list
|
Π‘ΠΏΠΈΡΠΎΠΊ Π²Π΅ΠΊΡΠΎΡΠΎΠ² ΠΏΡΠΈΠ·Π½Π°ΠΊΠΎΠ²/Π²ΠΊΡΠ°ΠΏΠ»Π΅Π½ΠΈΠΉ, ΠΊΠΎΡΠΎΡΡΠ΅ Π½ΡΠΆΠ½ΠΎ Π²Π΅ΡΠ½ΡΡΡ. |
None
|
ΠΠΎΠ·Π²ΡΠ°ΡΠ°Π΅ΡΡΡ:
Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ |
---|---|
Tensor
|
ΠΡΡ ΠΎΠ΄ ΠΌΠΎΠ΄Π΅Π»ΠΈ tensor. |
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
set_classes(text, batch=80, cache_clip_model=True)
ΠΠ°ΡΠ°Π½Π΅Π΅ ΡΡΡΠ°Π½ΠΎΠ²ΠΈ ΠΊΠ»Π°ΡΡΡ, ΡΡΠΎΠ±Ρ ΠΌΠΎΠ΄Π΅Π»Ρ ΠΌΠΎΠ³Π»Π° Π΄Π΅Π»Π°ΡΡ ΠΎΡΡΠ»Π°ΠΉΠ½-ΠΈΠ½ΡΠ΅ΡΠ΅Π½ΡΠΈΡ Π±Π΅Π· ΠΊΠ»ΠΈΠΏΠΎΠ²ΠΎΠΉ ΠΌΠΎΠ΄Π΅Π»ΠΈ.
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
ultralytics.nn.tasks.Ensemble
ΠΠ°Π·Ρ: ModuleList
ΠΠ½ΡΠ°ΠΌΠ±Π»Ρ ΠΌΠΎΠ΄Π΅Π»Π΅ΠΉ.
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
__init__()
forward(x, augment=False, profile=False, visualize=False)
Π€ΡΠ½ΠΊΡΠΈΡ Π³Π΅Π½Π΅ΡΠΈΡΡΠ΅Ρ ΠΏΠΎΡΠ»Π΅Π΄Π½ΠΈΠΉ ΡΠ»ΠΎΠΉ ΡΠ΅ΡΠΈ YOLO .
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
ultralytics.nn.tasks.temporary_modules(modules=None)
ΠΠΎΠ½ΡΠ΅ΠΊΡΡΠ½ΡΠΉ ΠΌΠ΅Π½Π΅Π΄ΠΆΠ΅Ρ Π΄Π»Ρ Π²ΡΠ΅ΠΌΠ΅Π½Π½ΠΎΠ³ΠΎ Π΄ΠΎΠ±Π°Π²Π»Π΅Π½ΠΈΡ ΠΈΠ»ΠΈ ΠΈΠ·ΠΌΠ΅Π½Π΅Π½ΠΈΡ ΠΌΠΎΠ΄ΡΠ»Π΅ΠΉ Π² ΠΌΠΎΠ΄ΡΠ»ΡΠ½ΠΎΠΌ ΠΊΡΡΠ΅ Python(sys.modules
).
ΠΡΡ ΡΡΠ½ΠΊΡΠΈΡ ΠΌΠΎΠΆΠ½ΠΎ ΠΈΡΠΏΠΎΠ»ΡΠ·ΠΎΠ²Π°ΡΡ Π΄Π»Ρ ΠΈΠ·ΠΌΠ΅Π½Π΅Π½ΠΈΡ ΠΏΡΡΠ΅ΠΉ ΠΊ ΠΌΠΎΠ΄ΡΠ»ΡΠΌ Π²ΠΎ Π²ΡΠ΅ΠΌΡ Π²ΡΠΏΠΎΠ»Π½Π΅Π½ΠΈΡ. ΠΡΠΎ ΠΏΠΎΠ»Π΅Π·Π½ΠΎ ΠΏΡΠΈ ΡΠ΅ΡΠ°ΠΊΡΠΎΡΠΈΠ½Π³Π΅ ΠΊΠΎΠ΄Π°, ΠΊΠΎΠ³Π΄Π° ΡΡ ΠΏΠ΅ΡΠ΅ΠΌΠ΅ΡΡΠΈΠ» ΠΌΠΎΠ΄ΡΠ»Ρ ΠΈΠ· ΠΎΠ΄Π½ΠΎΠ³ΠΎ ΠΌΠ΅ΡΡΠ° Π² Π΄ΡΡΠ³ΠΎΠ΅, Π½ΠΎ ΠΏΡΠΈ ΡΡΠΎΠΌ Ρ ΠΎΡΠ΅ΡΡ ΠΏΠΎΠ΄Π΄Π΅ΡΠΆΠΈΠ²Π°ΡΡ ΡΡΠ°ΡΡΠ΅ ΠΏΡΡΠΈ ΠΈΠΌΠΏΠΎΡΡΠ°. Π΄Π»Ρ ΠΎΠ±ΡΠ°ΡΠ½ΠΎΠΉ ΡΠΎΠ²ΠΌΠ΅ΡΡΠΈΠΌΠΎΡΡΠΈ.
ΠΠ°ΡΠ°ΠΌΠ΅ΡΡΡ:
ΠΠΌΡ | Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ | ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ |
---|---|---|---|
modules |
dict
|
Π‘Π»ΠΎΠ²Π°ΡΡ, ΠΎΡΠΎΠ±ΡΠ°ΠΆΠ°ΡΡΠΈΠΉ ΡΡΠ°ΡΡΠ΅ ΠΏΡΡΠΈ ΠΌΠΎΠ΄ΡΠ»Π΅ΠΉ Π½Π° Π½ΠΎΠ²ΡΠ΅. |
None
|
ΠΡΠΈΠΌΠ΅Ρ
ΠΡΠΈΠΌΠ΅ΡΠ°Π½ΠΈΠ΅
ΠΠ·ΠΌΠ΅Π½Π΅Π½ΠΈΡ Π΄Π΅ΠΉΡΡΠ²ΡΡΡ ΡΠΎΠ»ΡΠΊΠΎ Π²Π½ΡΡΡΠΈ ΠΊΠΎΠ½ΡΠ΅ΠΊΡΡΠ½ΠΎΠ³ΠΎ ΠΌΠ΅Π½Π΅Π΄ΠΆΠ΅ΡΠ° ΠΈ ΠΎΡΠΌΠ΅Π½ΡΡΡΡΡ ΠΏΠΎΡΠ»Π΅ Π΅Π³ΠΎ Π²ΡΡ
ΠΎΠ΄Π°.
ΠΠΌΠ΅ΠΉ Π² Π²ΠΈΠ΄Ρ, ΡΡΠΎ ΠΏΡΡΠΌΠΎΠ΅ ΠΌΠ°Π½ΠΈΠΏΡΠ»ΠΈΡΠΎΠ²Π°Π½ΠΈΠ΅ sys.modules
ΠΌΠΎΠΆΠ΅Ρ ΠΏΡΠΈΠ²Π΅ΡΡΠΈ ΠΊ Π½Π΅ΠΏΡΠ΅Π΄ΡΠΊΠ°Π·ΡΠ΅ΠΌΡΠΌ ΡΠ΅Π·ΡΠ»ΡΡΠ°ΡΠ°ΠΌ, ΠΎΡΠΎΠ±Π΅Π½Π½ΠΎ Π² Π±ΠΎΠ»ΡΡΠΈΡ
ΠΏΡΠΈΠ»ΠΎΠΆΠ΅Π½ΠΈΡΡ
ΠΈΠ»ΠΈ Π±ΠΈΠ±Π»ΠΈΠΎΡΠ΅ΠΊΠ°Ρ
. ΠΡΠΏΠΎΠ»ΡΠ·ΡΠΉ ΡΡΡ ΡΡΠ½ΠΊΡΠΈΡ Ρ ΠΎΡΡΠΎΡΠΎΠΆΠ½ΠΎΡΡΡΡ.
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
ultralytics.nn.tasks.torch_safe_load(weight)
ΠΡΠ° ΡΡΠ½ΠΊΡΠΈΡ ΠΏΡΡΠ°Π΅ΡΡΡ Π·Π°Π³ΡΡΠ·ΠΈΡΡ ΠΌΠΎΠ΄Π΅Π»Ρ PyTorch Ρ ΠΏΠΎΠΌΠΎΡΡΡ ΡΡΠ½ΠΊΡΠΈΠΈ torch.load(). ΠΡΠ»ΠΈ Π²ΠΎΠ·Π½ΠΈΠΊΠ°Π΅Ρ ΠΎΡΠΈΠ±ΠΊΠ° ModuleNotFoundError, ΠΎΠ½Π° ΠΏΠΎΠΉΠΌΠ°Π΅Ρ ΠΎΡΠΈΠ±ΠΊΡ, Π·Π°ΠΏΠΈΡΠ΅Ρ ΠΏΡΠ΅Π΄ΡΠΏΡΠ΅ΠΆΠ΄Π΅Π½ΠΈΠ΅ ΠΈ ΠΏΠΎΠΏΡΡΠ°Π΅ΡΡΡ ΡΡΡΠ°Π½ΠΎΠ²ΠΈΡΡ Π½Π΅Π΄ΠΎΡΡΠ°ΡΡΠΈΠΉ ΠΌΠΎΠ΄ΡΠ»Ρ Ρ ΠΏΠΎΠΌΠΎΡΡΡ ΡΡΠ½ΠΊΡΠΈΠΈ check_requirements(). ΠΠΎΡΠ»Π΅ ΡΡΡΠ°Π½ΠΎΠ²ΠΊΠΈ ΡΡΠ½ΠΊΡΠΈΡ ΡΠ½ΠΎΠ²Π° ΠΏΡΡΠ°Π΅ΡΡΡ Π·Π°Π³ΡΡΠ·ΠΈΡΡ ΠΌΠΎΠ΄Π΅Π»Ρ Ρ ΠΏΠΎΠΌΠΎΡΡΡ torch.load().
ΠΠ°ΡΠ°ΠΌΠ΅ΡΡΡ:
ΠΠΌΡ | Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ | ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ |
---|---|---|---|
weight |
str
|
ΠΡΡΡ ΠΊ ΡΠ°ΠΉΠ»Ρ Ρ ΠΌΠΎΠ΄Π΅Π»ΡΡ PyTorch . |
ΡΡΠ΅Π±ΡΠ΅ΡΡΡ |
ΠΠΎΠ·Π²ΡΠ°ΡΠ°Π΅ΡΡΡ:
Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ |
---|---|
dict
|
ΠΠ°Π³ΡΡΠΆΠ΅Π½Π½Π°Ρ ΠΌΠΎΠ΄Π΅Π»Ρ PyTorch . |
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
ultralytics.nn.tasks.attempt_load_weights(weights, device=None, inplace=True, fuse=False)
ΠΠ°Π³ΡΡΠΆΠ°Π΅Ρ Π°Π½ΡΠ°ΠΌΠ±Π»Ρ ΠΌΠΎΠ΄Π΅Π»Π΅ΠΉ weights=[a,b,c] ΠΈΠ»ΠΈ ΠΎΠ΄Π½Ρ ΠΌΠΎΠ΄Π΅Π»Ρ weights=[a] ΠΈΠ»ΠΈ weights=a.
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
ultralytics.nn.tasks.attempt_load_one_weight(weight, device=None, inplace=True, fuse=False)
ΠΠ°Π³ΡΡΠΆΠ°Π΅Ρ ΠΎΠ΄Π½Ρ ΠΌΠΎΠ΄Π΅Π»Ρ Π²Π΅ΡΠΎΠ².
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
ultralytics.nn.tasks.parse_model(d, ch, verbose=True)
Π Π°Π·Π±Π΅ΡΠΈ ΡΠ»ΠΎΠ²Π°ΡΡ YOLO model.yaml Π½Π° ΠΌΠΎΠ΄Π΅Π»ΠΈ PyTorch .
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 |
|
ultralytics.nn.tasks.yaml_model_load(path)
ΠΠ°Π³ΡΡΠ·ΠΈ ΠΌΠΎΠ΄Π΅Π»Ρ YOLOv8 ΠΈΠ· ΡΠ°ΠΉΠ»Π° YAML.
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
ultralytics.nn.tasks.guess_model_scale(model_path)
ΠΡΠΈΠ½ΠΈΠΌΠ°Π΅Ρ Π² ΠΊΠ°ΡΠ΅ΡΡΠ²Π΅ Π²Ρ ΠΎΠ΄Π½ΡΡ Π΄Π°Π½Π½ΡΡ ΠΏΡΡΡ ΠΊ YAML-ΡΠ°ΠΉΠ»Ρ ΠΌΠΎΠ΄Π΅Π»ΠΈ YOLO ΠΈ ΠΈΠ·Π²Π»Π΅ΠΊΠ°Π΅Ρ ΡΠΈΠΌΠ²ΠΎΠ» ΡΠ°Π·ΠΌΠ΅ΡΠ° ΠΈΠ· ΠΌΠ°ΡΡΡΠ°Π±Π° ΠΌΠΎΠ΄Π΅Π»ΠΈ. Π€ΡΠ½ΠΊΡΠΈΡ ΠΈΡΠΏΠΎΠ»ΡΠ·ΡΠ΅Ρ ΡΠΎΠΏΠΎΡΡΠ°Π²Π»Π΅Π½ΠΈΠ΅ ΡΠ΅Π³ΡΠ»ΡΡΠ½ΡΡ Π²ΡΡΠ°ΠΆΠ΅Π½ΠΈΠΉ Π΄Π»Ρ ΠΏΠΎΠΈΡΠΊΠ° ΡΠ°Π±Π»ΠΎΠ½Π° ΠΌΠ°ΡΡΡΠ°Π±Π° ΠΌΠΎΠ΄Π΅Π»ΠΈ Π² ΠΈΠΌΠ΅Π½ΠΈ YAML-ΡΠ°ΠΉΠ»Π°, ΠΊΠΎΡΠΎΡΡΠΉ ΠΎΠ±ΠΎΠ·Π½Π°ΡΠ°Π΅ΡΡΡ ΡΠΈΠΌΠ²ΠΎΠ»Π°ΠΌΠΈ n, s, m, l ΠΈΠ»ΠΈ x. Π€ΡΠ½ΠΊΡΠΈΡ Π²ΠΎΠ·Π²ΡΠ°ΡΠ°Π΅Ρ ΡΠΈΠΌΠ²ΠΎΠ» ΡΠ°Π·ΠΌΠ΅ΡΠ° ΡΠΊΠ°Π»Ρ ΠΌΠΎΠ΄Π΅Π»ΠΈ Π² Π²ΠΈΠ΄Π΅ ΡΡΡΠΎΠΊΠΈ.
ΠΠ°ΡΠ°ΠΌΠ΅ΡΡΡ:
ΠΠΌΡ | Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ | ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ |
---|---|---|---|
model_path |
str | Path
|
ΠΡΡΡ ΠΊ YAML-ΡΠ°ΠΉΠ»Ρ ΠΌΠΎΠ΄Π΅Π»ΠΈ YOLO . |
ΡΡΠ΅Π±ΡΠ΅ΡΡΡ |
ΠΠΎΠ·Π²ΡΠ°ΡΠ°Π΅ΡΡΡ:
Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ |
---|---|
str
|
Π Π°Π·ΠΌΠ΅ΡΠ½ΡΠΉ ΡΠΈΠΌΠ²ΠΎΠ» ΠΌΠ°ΡΡΡΠ°Π±Π° ΠΌΠΎΠ΄Π΅Π»ΠΈ, ΠΊΠΎΡΠΎΡΡΠΉ ΠΌΠΎΠΆΠ΅Ρ Π±ΡΡΡ n, s, m, l ΠΈΠ»ΠΈ x. |
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
ultralytics.nn.tasks.guess_model_task(model)
Π£Π³Π°Π΄Π°ΠΉ Π·Π°Π΄Π°ΡΡ ΠΌΠΎΠ΄Π΅Π»ΠΈ PyTorch ΠΏΠΎ Π΅Π΅ Π°ΡΡ ΠΈΡΠ΅ΠΊΡΡΡΠ΅ ΠΈΠ»ΠΈ ΠΊΠΎΠ½ΡΠΈΠ³ΡΡΠ°ΡΠΈΠΈ.
ΠΠ°ΡΠ°ΠΌΠ΅ΡΡΡ:
ΠΠΌΡ | Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ | ΠΠΎ ΡΠΌΠΎΠ»ΡΠ°Π½ΠΈΡ |
---|---|---|---|
model |
Module | dict
|
PyTorch ΠΌΠΎΠ΄Π΅Π»Ρ ΠΈΠ»ΠΈ ΠΊΠΎΠ½ΡΠΈΠ³ΡΡΠ°ΡΠΈΡ ΠΌΠΎΠ΄Π΅Π»ΠΈ Π² ΡΠΎΡΠΌΠ°ΡΠ΅ YAML. |
ΡΡΠ΅Π±ΡΠ΅ΡΡΡ |
ΠΠΎΠ·Π²ΡΠ°ΡΠ°Π΅ΡΡΡ:
Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ |
---|---|
str
|
ΠΠ°Π΄Π°ΡΠ° ΠΌΠΎΠ΄Π΅Π»ΠΈ ("ΠΎΠ±Π½Π°ΡΡΠΆΠΈΡΡ", "ΡΠ΅Π³ΠΌΠ΅Π½ΡΠΈΡΠΎΠ²Π°ΡΡ", "ΠΊΠ»Π°ΡΡΠΈΡΠΈΡΠΈΡΠΎΠ²Π°ΡΡ", "ΠΏΠΎΠ·ΠΈΡΠΎΠ²Π°ΡΡ"). |
ΠΠΎΠ΄Π½ΠΈΠΌΠ°Π΅Ρ:
Π’ΠΈΠΏ | ΠΠΏΠΈΡΠ°Π½ΠΈΠ΅ |
---|---|
SyntaxError
|
ΠΡΠ»ΠΈ Π·Π°Π΄Π°ΡΡ ΠΌΠΎΠ΄Π΅Π»ΠΈ ΠΎΠΏΡΠ΅Π΄Π΅Π»ΠΈΡΡ Π½Π΅ ΡΠ΄Π°Π»ΠΎΡΡ. |
ΠΡΡ
ΠΎΠ΄Π½ΡΠΉ ΠΊΠΎΠ΄ Π² ultralytics/nn/tasks.py
982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 |
|
Π‘ΠΎΠ·Π΄Π°Π½ΠΎ 2023-11-12, ΠΠ±Π½ΠΎΠ²Π»Π΅Π½ΠΎ 2024-03-03
ΠΠ²ΡΠΎΡΡ: glenn-jocher (6), Laughing-q (1)