рдХреЗ рд▓рд┐рдП рд╕рдВрджрд░реНрдн ultralytics/nn/tasks.py
рдиреЛрдЯ
рдпрд╣ рдлрд╝рд╛рдЗрд▓ рдпрд╣рд╛рдБ рдЙрдкрд▓рдмреНрдз рд╣реИ https://github.com/ultralytics/ultralytics/рдмреВрдБрдж/рдореБрдЦреНрдп/ultralytics/nn/tasks.py рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░реЗрдВред рдпрджрд┐ рдЖрдк рдХреЛрдИ рд╕рдорд╕реНрдпрд╛ рджреЗрдЦрддреЗ рд╣реИрдВ рддреЛ рдХреГрдкрдпрд╛ рдкреБрд▓ рдЕрдиреБрд░реЛрдз рдХрд╛ рдпреЛрдЧрджрд╛рди рдХрд░рдХреЗ рдЗрд╕реЗ рдареАрдХ рдХрд░рдиреЗ рдореЗрдВ рдорджрдж рдХрд░реЗрдВ ЁЯЫая╕Пред ЁЯЩП рдзрдиреНрдпрд╡рд╛рдж !
ultralytics.nn.tasks.BaseModel
рдХрд╛ рд░реВрдк: Module
рдмреЗрд╕рдореЛрдбреЗрд▓ рдХреНрд▓рд╛рд╕ рд╕рднреА рдореЙрдбрд▓реЛрдВ рдХреЗ рд▓рд┐рдП рдмреЗрд╕ рдХреНрд▓рд╛рд╕ рдХреЗ рд░реВрдк рдореЗрдВ рдХрд╛рд░реНрдп рдХрд░рддрд╛ рд╣реИ 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 рдФрд░ рдЬреАрдЯреА рд▓реЗрдмрд▓ред |
рдЖрд╡рд╢реНрдпрдХ |
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
Tensor
|
рдиреЗрдЯрд╡рд░реНрдХ рдХрд╛ рдЖрдЙрдЯрдкреБрдЯ. |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/tasks.py
fuse(verbose=True)
рдлреНрдпреВрдЬ Conv2d()
рдФрд░ BatchNorm2d()
рдореЙрдбрд▓ рдХреА рдкрд░рддреЛрдВ рдХреЛ рдПрдХ рдкрд░рдд рдореЗрдВ рдмрджрд▓рдиреЗ рдХреЗ рд▓рд┐рдП,
рдЧрдгрдирд╛ рджрдХреНрд╖рддрд╛ред
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
Module
|
рдлреНрдпреВрдЬреНрдб рдореЙрдбрд▓ рд╡рд╛рдкрд╕ рдЖ рдЧрдпрд╛ рд╣реИред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/tasks.py
info(detailed=False, verbose=True, imgsz=640)
рдореЙрдбрд▓ рдХреА рдЬрд╛рдирдХрд╛рд░реА рдкреНрд░рд┐рдВрдЯ рдХрд░рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
detailed |
bool
|
рдпрджрд┐ рд╕рд╣реА рд╣реИ, рддреЛ рдореЙрдбрд▓ рдХреЗ рдмрд╛рд░реЗ рдореЗрдВ рд╡рд┐рд╕реНрддреГрдд рдЬрд╛рдирдХрд╛рд░реА рдкреНрд░рд┐рдВрдЯ рдХрд░рддрд╛ рд╣реИред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рд╕реЗ рдЧрд▓рдд |
False
|
verbose |
bool
|
рдпрджрд┐ рд╕рд╣реА рд╣реИ, рддреЛ рдореЙрдбрд▓ рдХреА рдЬрд╛рдирдХрд╛рд░реА рдкреНрд░рд┐рдВрдЯ рдХрд░рддрд╛ рд╣реИред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рд╕реЗ рдЧрд▓рдд |
True
|
imgsz |
int
|
рдЫрд╡рд┐ рдХрд╛ рдЖрдХрд╛рд░ рдЬрд┐рд╕ рдкрд░ рдореЙрдбрд▓ рдХреЛ рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рдХрд┐рдпрд╛ рдЬрд╛рдПрдЧрд╛ред 640 рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯ |
640
|
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/tasks.py
init_criterion()
BaseModel рдХреЗ рд▓рд┐рдП рд╣рд╛рдирд┐ рдорд╛рдирджрдВрдб рдкреНрд░рд╛рд░рдВрдн рдХрд░реЗрдВред
is_fused(thresh=10)
рдЬрд╛рдВрдЪреЗрдВ рдХрд┐ рдХреНрдпрд╛ рдореЙрдбрд▓ рдореЗрдВ BatchNorm рдкрд░рддреЛрдВ рдХреА рдПрдХ рдирд┐рд╢реНрдЪрд┐рдд рд╕реАрдорд╛ рд╕реЗ рдХрдо рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
thresh |
int
|
BatchNorm рдкрд░рддреЛрдВ рдХреА рдереНрд░реЗрд╢реЛрд▓реНрдб рд╕рдВрдЦреНрдпрд╛ред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 10 рд╣реИред |
10
|
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
bool
|
рд╕рдЪ рд╣реИ рдЕрдЧрд░ рдореЙрдбрд▓ рдореЗрдВ BatchNorm рдкрд░рддреЛрдВ рдХреА рд╕рдВрдЦреНрдпрд╛ рд╕реАрдорд╛ рд╕реЗ рдХрдо рд╣реИ, рдЕрдиреНрдпрдерд╛ рдЧрд▓рддред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/tasks.py
load(weights, verbose=True)
рдореЙрдбрд▓ рдореЗрдВ рд╡рдЬрди рд▓реЛрдб рдХрд░реЗрдВред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
weights |
dict | Module
|
рд▓реЛрдб рдХрд┐рдП рдЬрд╛рдиреЗ рд╡рд╛рд▓реЗ рдкреВрд░реНрд╡-рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рднрд╛рд░ред |
рдЖрд╡рд╢реНрдпрдХ |
verbose |
bool
|
рд╕реНрдерд╛рдирд╛рдВрддрд░рдг рдкреНрд░рдЧрддрд┐ рдХреЛ рд▓реЙрдЧ рдХрд░рдирд╛ рд╣реИ рдпрд╛ рдирд╣реАрдВред рд╕рд╣реА рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
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
|
visualize |
bool
|
рдореЙрдбрд▓ рдХреЗ рдлреАрдЪрд░ рдореИрдкреНрд╕ рдХреЛ рд╕реЗрд╡ рдХрд░реЗрдВ рдпрджрд┐ True, рдбрд┐рдлрд╝реЙрд▓реНрдЯ рд░реВрдк рд╕реЗ рдЧрд▓рдд рд╣реИред |
False
|
augment |
bool
|
рдкреВрд░реНрд╡рд╛рдиреБрдорд╛рди рдХреЗ рджреМрд░рд╛рди рдЫрд╡рд┐ рдХреЛ рдмрдврд╝рд╛рдПрдБ, рдбрд┐рдлрд╝реЙрд▓реНрдЯ рд░реВрдк рд╕реЗ рдЧрд▓рдд рд╣реЛ рдЬрд╛рддрд╛ рд╣реИ. |
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
init_criterion()
ultralytics.nn.tasks.OBBModel
рдХрд╛ рд░реВрдк: DetectionModel
YOLOv8 рдУрд░рд┐рдПрдВрдЯреЗрдб рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ (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 рджрд┐рдП рдЧрдП рдХреЙрдиреНрдлрд╝рд┐рдЧрд░реЗрд╢рди рдФрд░ рдорд╛рдкрджрдВрдбреЛрдВ рдХреЗ рд╕рд╛рде рд╡рд┐рднрд╛рдЬрди рдореЙрдбрд▓ред
init_criterion()
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 рдкреЛрдЬрд╝ рдореЙрдбрд▓ред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/tasks.py
init_criterion()
ultralytics.nn.tasks.ClassificationModel
рдХрд╛ рд░реВрдк: BaseModel
YOLOv8 рд╡рд░реНрдЧреАрдХрд░рдг рдореЙрдбрд▓ред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/tasks.py
__init__(cfg='yolov8n-cls.yaml', ch=3, nc=None, verbose=True)
рд╡рд╛рдИрдПрдПрдордПрд▓ рдХреЗ рд╕рд╛рде рдЗрдирд┐рдЯ рд╡рд░реНрдЧреАрдХрд░рдг, рдЪреИрдирд▓, рдХрдХреНрд╖рд╛рдУрдВ рдХреА рд╕рдВрдЦреНрдпрд╛, рд╡рд░реНрдмреЛрдЬрд╝ рдзреНрд╡рдЬред
init_criterion()
reshape_outputs(model, nc)
staticmethod
рдпрджрд┐ рдЖрд╡рд╢реНрдпрдХ рд╣реЛ рддреЛ рд╡рд░реНрдЧ рдЧрдгрдирд╛ 'рдПрди' рдХреЗ рд▓рд┐рдП рдПрдХ TorchVision рд╡рд░реНрдЧреАрдХрд░рдг рдореЙрдбрд▓ рдЕрджреНрдпрддрди рдХрд░реЗрдВ.
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/tasks.py
ultralytics.nn.tasks.RTDETRDetectionModel
рдХрд╛ рд░реВрдк: DetectionModel
RTDETR (рд░реАрдпрд▓-рдЯрд╛рдЗрдо рдбрд┐рдЯреЗрдХреНрд╢рди рдФрд░ рдЯреНрд░рд╛рдВрд╕рдлреЙрд░реНрдорд░ рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░рдХреЗ рдЯреНрд░реИрдХрд┐рдВрдЧ) рдбрд┐рдЯреЗрдХреНрд╢рди рдореЙрдбрд▓ рдХреНрд▓рд╛рд╕ред
рдпрд╣ рд╡рд░реНрдЧ RTDETR рдЖрд░реНрдХрд┐рдЯреЗрдХреНрдЪрд░ рдХреЗ рдирд┐рд░реНрдорд╛рдг, рд╣рд╛рдирд┐ рдХрд╛рд░реНрдпреЛрдВ рдХреЛ рдкрд░рд┐рднрд╛рд╖рд┐рдд рдХрд░рдиреЗ рдФрд░ рджреЛрдиреЛрдВ рдХреЛ рд╕реБрд╡рд┐рдзрд╛рдЬрдирдХ рдмрдирд╛рдиреЗ рдХреЗ рд▓рд┐рдП рдЬрд┐рдореНрдореЗрджрд╛рд░ рд╣реИ рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдФрд░ рдЕрдиреБрдорд╛рди рдкреНрд░рдХреНрд░рд┐рдпрд╛рдПрдВред RTDETR рдПрдХ рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди рдФрд░ рдЯреНрд░реИрдХрд┐рдВрдЧ рдореЙрдбрд▓ рд╣реИ рдЬреЛ рдбрд┐рдЯреЗрдХреНрд╢рдирдореЙрдбрд▓ рдмреЗрд╕ рдХреНрд▓рд╛рд╕ред
рд╡рд┐рд╢реЗрд╖рддрд╛рдПрдБ:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|---|
cfg |
str
|
рдХреЙрдиреНрдлрд╝рд┐рдЧрд░реЗрд╢рди рдлрд╝рд╛рдЗрд▓ рдкрде рдпрд╛ рдкреНрд░реАрд╕реЗрдЯ рд╕реНрдЯреНрд░рд┐рдВрдЧред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 'rtdetr-l.yaml' рд╣реИред |
ch |
int
|
рдЗрдирдкреБрдЯ рдЪреИрдирд▓реЛрдВ рдХреА рд╕рдВрдЦреНрдпрд╛ред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 3 (RGB) рд╣реИред |
nc |
int
|
рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди рдХреЗ рд▓рд┐рдП рдХрдХреНрд╖рд╛рдУрдВ рдХреА рд╕рдВрдЦреНрдпрд╛ред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рдХреЛрдИ рдирд╣реАрдВ рд╣реИред |
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
|
verbose |
bool
|
рдЖрд░рдВрднреАрдХрд░рдг рдХреЗ рджреМрд░рд╛рди рдЕрддрд┐рд░рд┐рдХреНрдд рдЬрд╛рдирдХрд╛рд░реА рдкреНрд░рд┐рдВрдЯ рдХрд░реЗрдВред рд╕рд╣реА рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
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
|
рдпрджрд┐ рд╕рд╣реА рд╣реИ, рддреЛ рдкреНрд░рддреНрдпреЗрдХ рдкрд░рдд рдХреЗ рд▓рд┐рдП рдЧрдгрдирд╛ рд╕рдордп рдХреЛ рдкреНрд░реЛрдлрд╛рдЗрд▓ рдХрд░реЗрдВред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рд░реВрдк рд╕реЗ рдЧрд▓рдд рд╣реИ. |
False
|
visualize |
bool
|
рдпрджрд┐ рд╕рд╣реА рд╣реИ, рддреЛ рд╡рд┐рдЬрд╝реБрдЕрд▓рд╛рдЗрдЬрд╝реЗрд╢рди рдХреЗ рд▓рд┐рдП рд╕реБрд╡рд┐рдзрд╛ рдорд╛рдирдЪрд┐рддреНрд░ рд╕рд╣реЗрдЬреЗрдВред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рд░реВрдк рд╕реЗ рдЧрд▓рдд рд╣реИ. |
False
|
batch |
dict
|
рдореВрд▓реНрдпрд╛рдВрдХрди рдХреЗ рд▓рд┐рдП рдЬрдореАрдиреА рд╕рдЪреНрдЪрд╛рдИ рдбреЗрдЯрд╛ред рдХреЛрдИ рдирд╣реАрдВ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
None
|
augment |
bool
|
рдпрджрд┐ рд╕рд╣реА рд╣реИ, рддреЛ рдЕрдиреБрдорд╛рди рдХреЗ рджреМрд░рд╛рди рдбреЗрдЯрд╛ рд╡реГрджреНрдзрд┐ рдХрд░реЗрдВред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рд░реВрдк рд╕реЗ рдЧрд▓рдд рд╣реИ. |
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
|
рдпрджрд┐ рд╕рд╣реА рд╣реИ, рддреЛ рдкреНрд░рддреНрдпреЗрдХ рдкрд░рдд рдХреЗ рд▓рд┐рдП рдЧрдгрдирд╛ рд╕рдордп рдХреЛ рдкреНрд░реЛрдлрд╛рдЗрд▓ рдХрд░реЗрдВред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рд░реВрдк рд╕реЗ рдЧрд▓рдд рд╣реИ. |
False
|
visualize |
bool
|
рдпрджрд┐ рд╕рд╣реА рд╣реИ, рддреЛ рд╡рд┐рдЬрд╝реБрдЕрд▓рд╛рдЗрдЬрд╝реЗрд╢рди рдХреЗ рд▓рд┐рдП рд╕реБрд╡рд┐рдзрд╛ рдорд╛рдирдЪрд┐рддреНрд░ рд╕рд╣реЗрдЬреЗрдВред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рд░реВрдк рд╕реЗ рдЧрд▓рдд рд╣реИ. |
False
|
txt_feats |
Tensor
|
рдкрд╛рда рд╕реБрд╡рд┐рдзрд╛рдПрдБ, рдпрджрд┐ рдпрд╣ рджрд┐рдпрд╛ рдЧрдпрд╛ рд╣реИ рддреЛ рдЗрд╕рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░реЗрдВред рдХреЛрдИ рдирд╣реАрдВ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
None
|
augment |
bool
|
рдпрджрд┐ рд╕рд╣реА рд╣реИ, рддреЛ рдЕрдиреБрдорд╛рди рдХреЗ рджреМрд░рд╛рди рдбреЗрдЯрд╛ рд╡реГрджреНрдзрд┐ рдХрд░реЗрдВред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рд░реВрдк рд╕реЗ рдЧрд▓рдд рд╣реИ. |
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)
рдореЙрдбрд▓ рд╡рдЬрди рдХрд╛ рдПрдХ рдкрд╣рдирд╛рд╡рд╛ рд▓реЛрдб рдХрд░рддрд╛ рд╣реИ = [рдП, рдмреА, рд╕реА] рдпрд╛ рдПрдХ рдПрдХрд▓ рдореЙрдбрд▓ рд╡рдЬрди = [рдП] рдпрд╛ рд╡рдЬрди = рдПред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб 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
830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900 901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942 |
|
ultralytics.nn.tasks.yaml_model_load(path)
рдПрдХ рд▓реЛрдб рдХрд░реЗрдВ YOLOv8 рдПрдХ YAML рдлрд╝рд╛рдЗрд▓ рд╕реЗ рдореЙрдбрд▓ред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/tasks.py
ultralytics.nn.tasks.guess_model_scale(model_path)
рдПрдХ рдХреЗ рд▓рд┐рдП рдПрдХ рд░рд╛рд╕реНрддрд╛ рд▓реЗрддрд╛ рд╣реИ YOLO рдЗрдирдкреБрдЯ рдХреЗ рд░реВрдк рдореЗрдВ рдореЙрдбрд▓ рдХреА YAML рдлрд╝рд╛рдЗрд▓ рдФрд░ рдореЙрдбрд▓ рдХреЗ рдкреИрдорд╛рдиреЗ рдХреЗ рдЖрдХрд╛рд░ рдЪрд░рд┐рддреНрд░ рдХреЛ рдирд┐рдХрд╛рд▓рддреА рд╣реИред рд╕рдорд╛рд░реЛрд╣ YAML рдлрд╝рд╛рдЗрд▓ рдирд╛рдо рдореЗрдВ рдореЙрдбрд▓ рд╕реНрдХреЗрд▓ рдХреЗ рдкреИрдЯрд░реНрди рдХреЛ рдЦреЛрдЬрдиреЗ рдХреЗ рд▓рд┐рдП рдирд┐рдпрдорд┐рдд рдЕрднрд┐рд╡реНрдпрдХреНрддрд┐ рдорд┐рд▓рд╛рди рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░рддрд╛ рд╣реИ, рдЬрд┐рд╕реЗ рджреНрд╡рд╛рд░рд╛ рджрд░реНрд╢рд╛рдпрд╛ рдЬрд╛рддрд╛ рд╣реИ рдПрди, рдПрд╕, рдПрдо, рдПрд▓, рдпрд╛ рдПрдХреНрд╕ред рдлрд╝рдВрдХреНрд╢рди рд╕реНрдЯреНрд░рд┐рдВрдЧ рдХреЗ рд░реВрдк рдореЗрдВ рдореЙрдбрд▓ рд╕реНрдХреЗрд▓ рдХреЗ рдЖрдХрд╛рд░ рдЪрд░рд┐рддреНрд░ рдХреЛ рд▓реМрдЯрд╛рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
model_path |
str | Path
|
рдХреЗ рд▓рд┐рдП рдкрде YOLO рдореЙрдбрд▓ рдХреА YAML рдлрд╝рд╛рдЗрд▓ред |
рдЖрд╡рд╢реНрдпрдХ |
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
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
рд▓реЗрдЦрдХ: рдЧреНрд▓реЗрди-рдЬреЛрдЪрд░ (6), рд▓рд╛рдлрд┐рдВрдЧ-рдХреНрдпреВ (1)