рдХреЗ рд▓рд┐рдП рд╕рдВрджрд░реНрдн ultralytics/nn/modules/head.py
рдиреЛрдЯ
рдпрд╣ рдлрд╝рд╛рдЗрд▓ рдпрд╣рд╛рдБ рдЙрдкрд▓рдмреНрдз рд╣реИ https://github.com/ultralytics/ultralytics/рдмреВрдБрдж/рдореБрдЦреНрдп/ultralytics/nn/modules/head.py. рдпрджрд┐ рдЖрдк рдХреЛрдИ рд╕рдорд╕реНрдпрд╛ рджреЗрдЦрддреЗ рд╣реИрдВ рддреЛ рдХреГрдкрдпрд╛ рдкреБрд▓ рдЕрдиреБрд░реЛрдз рдХрд╛ рдпреЛрдЧрджрд╛рди рдХрд░рдХреЗ рдЗрд╕реЗ рдареАрдХ рдХрд░рдиреЗ рдореЗрдВ рдорджрдж рдХрд░реЗрдВ ЁЯЫая╕Пред ЁЯЩП рдзрдиреНрдпрд╡рд╛рдж !
ultralytics.nn.modules.head.Detect
рдХрд╛ рд░реВрдк: Module
YOLOv8 рдкрд╣рдЪрд╛рди рдореЙрдбрд▓ рдХреЗ рд▓рд┐рдП рд╕рд┐рд░ рдХрд╛ рдкрддрд╛ рд▓рдЧрд╛рдПрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/modules/head.py
__init__(nc=80, ch=())
рдЗрдирдХреНрд╡рд┐рдпрд▓рд╛рдЗрдЬрд╝ рдХрд░рддрд╛ рд╣реИ YOLOv8 рдХрдХреНрд╖рд╛рдУрдВ рдФрд░ рдЪреИрдирд▓реЛрдВ рдХреА рдирд┐рд░реНрджрд┐рд╖реНрдЯ рд╕рдВрдЦреНрдпрд╛ рдХреЗ рд╕рд╛рде рдбрд┐рдЯреЗрдХреНрд╢рди рдкрд░рддред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/modules/head.py
bias_init()
рдбрд┐рдЯреЗрдХреНрдЯ() рдкреВрд░реНрд╡рд╛рдЧреНрд░рд╣реЛрдВ рдХреЛ рдкреНрд░рд╛рд░рдВрдн рдХрд░реЗрдВ, рдЪреЗрддрд╛рд╡рдиреА: рд╕реНрдЯреНрд░рд╛рдЗрдб рдЙрдкрд▓рдмреНрдзрддрд╛ рдХреА рдЖрд╡рд╢реНрдпрдХрддрд╛ рд╣реИред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/modules/head.py
decode_bboxes(bboxes, anchors)
forward(x)
рдЕрдиреБрдорд╛рдирд┐рдд рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдФрд░ рд╡рд░реНрдЧ рд╕рдВрднрд╛рд╡рдирд╛рдУрдВ рдХреЛ рдЬреЛрдбрд╝рддрд╛ рд╣реИ рдФрд░ рд▓реМрдЯрд╛рддрд╛ рд╣реИред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/modules/head.py
ultralytics.nn.modules.head.Segment
рдХрд╛ рд░реВрдк: Detect
YOLOv8 рд╡рд┐рднрд╛рдЬрди рдореЙрдбрд▓ рдХреЗ рд▓рд┐рдП рд╕реЗрдЧрдореЗрдВрдЯ рд╣реЗрдбред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/modules/head.py
__init__(nc=80, nm=32, npr=256, ch=())
рдЗрдирд░рд┐рдЬрд┐рдирд┐рдпрд▓рд╛рдЗрдЬрд╝ рдХрд░реЗрдВ YOLO рдореЙрдбрд▓ рд╡рд┐рд╢реЗрд╖рддрд╛рдПрдБ рдЬреИрд╕реЗ рдорд╛рд╕реНрдХ рдХреА рд╕рдВрдЦреНрдпрд╛, рдкреНрд░реЛрдЯреЛрдЯрд╛рдЗрдк рдФрд░ рдХрдирд╡рд▓реНрд╢рди рдкрд░рддреЗрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/modules/head.py
forward(x)
рдпрджрд┐ рдкреНрд░рд╢рд┐рдХреНрд╖рдг рд╣реИ рддреЛ рдореЙрдбрд▓ рдЖрдЙрдЯрдкреБрдЯ рдФрд░ рдорд╛рд╕реНрдХ рдЧреБрдгрд╛рдВрдХ рд▓реМрдЯрд╛рдПрдВ, рдЕрдиреНрдпрдерд╛ рдЖрдЙрдЯрдкреБрдЯ рдФрд░ рдорд╛рд╕реНрдХ рдЧреБрдгрд╛рдВрдХ рд▓реМрдЯрд╛рдПрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/modules/head.py
ultralytics.nn.modules.head.OBB
рдХрд╛ рд░реВрдк: Detect
YOLOv8 рд░реЛрдЯреЗрд╢рди рдореЙрдбрд▓ рдХреЗ рд╕рд╛рде рдкрддрд╛ рд▓рдЧрд╛рдиреЗ рдХреЗ рд▓рд┐рдП OBB рдбрд┐рдЯреЗрдХреНрд╢рди рд╣реЗрдбред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/modules/head.py
__init__(nc=80, ne=1, ch=())
рдХрдХреНрд╖рд╛рдУрдВ рдХреА рд╕рдВрдЦреНрдпрд╛ рдХреЗ рд╕рд╛рде OBB рдкреНрд░рд╛рд░рдВрдн рдХрд░реЗрдВ nc
рдФрд░ рдкрд░рдд рдЪреИрдирд▓ ch
.
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/modules/head.py
decode_bboxes(bboxes, anchors)
рдШреБрдорд╛рдП рдЧрдП рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдХреЛ рдбреАрдХреЛрдб рдХрд░реЗрдВред
forward(x)
рдЕрдиреБрдорд╛рдирд┐рдд рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдФрд░ рд╡рд░реНрдЧ рд╕рдВрднрд╛рд╡рдирд╛рдУрдВ рдХреЛ рдЬреЛрдбрд╝рддрд╛ рд╣реИ рдФрд░ рд▓реМрдЯрд╛рддрд╛ рд╣реИред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/modules/head.py
ultralytics.nn.modules.head.Pose
рдХрд╛ рд░реВрдк: Detect
YOLOv8 рдХреАрдкреЙрдЗрдВрдЯ рдореЙрдбрд▓ рдХреЗ рд▓рд┐рдП рдкреЛрдЬрд╝ рд╣реЗрдбред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/modules/head.py
__init__(nc=80, kpt_shape=(17, 3), ch=())
рдкреНрд░рд╛рд░рдВрдн YOLO рдбрд┐рдлрд╝реЙрд▓реНрдЯ рдорд╛рдкрджрдВрдбреЛрдВ рдФрд░ рджреГрдврд╝ рдкрд░рддреЛрдВ рдХреЗ рд╕рд╛рде рдиреЗрдЯрд╡рд░реНрдХред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/modules/head.py
forward(x)
рдлреЙрд░рд╡рд░реНрдб рдкрд╛рд╕ рдереНрд░реВ рдХрд░реЗрдВ YOLO рдореЙрдбрд▓ рдФрд░ рд╡рд╛рдкрд╕реА рднрд╡рд┐рд╖реНрдпрд╡рд╛рдгрд┐рдпрд╛рдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/modules/head.py
kpts_decode(bs, kpts)
рдХреАрдкреЙрдЗрдВрдЯреНрд╕ рдХреЛ рдбреАрдХреЛрдб рдХрд░рддрд╛ рд╣реИред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/modules/head.py
ultralytics.nn.modules.head.Classify
рдХрд╛ рд░реВрдк: Module
YOLOv8 рд╡рд░реНрдЧреАрдХрд░рдг рд╢реАрд░реНрд╖, рдпрд╛рдиреА рдПрдХреНрд╕ (рдмреА, рд╕реА 1,20,20) рд╕реЗ рдПрдХреНрд╕ (рдмреА, рд╕реА 2)ред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/modules/head.py
__init__(c1, c2, k=1, s=1, p=None, g=1)
рдкреНрд░рд╛рд░рдВрдн рдХрд░рддрд╛ рд╣реИ YOLOv8 рдирд┐рд░реНрджрд┐рд╖реНрдЯ рдЗрдирдкреБрдЯ рдФрд░ рдЖрдЙрдЯрдкреБрдЯ рдЪреИрдирд▓реЛрдВ рдХреЗ рд╕рд╛рде рд╡рд░реНрдЧреАрдХрд░рдг рд╕рд┐рд░, рдХрд░реНрдиреЗрд▓ рдЖрдХрд╛рд░, рд╕реНрдЯреНрд░рд╛рдЗрдб, рдкреИрдбрд┐рдВрдЧ, рдФрд░ рд╕рдореВрд╣ред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/modules/head.py
forward(x)
рдХрд╛ рдлреЙрд░рд╡рд░реНрдб рдкрд╛рд╕ рдХрд░рддрд╛ рд╣реИ YOLO рдЗрдирдкреБрдЯ рдЫрд╡рд┐ рдбреЗрдЯрд╛ рдкрд░ рдореЙрдбрд▓ред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/modules/head.py
ultralytics.nn.modules.head.WorldDetect
рдХрд╛ рд░реВрдк: Detect
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/modules/head.py
__init__(nc=80, embed=512, with_bn=False, ch=())
рдкреНрд░рд╛рд░рдВрдн YOLOv8 рдПрдирд╕реА рдХрдХреНрд╖рд╛рдУрдВ рдФрд░ рдкрд░рдд рдЪреИрдирд▓реЛрдВ рд╕реАрдПрдЪ рдХреЗ рд╕рд╛рде рдбрд┐рдЯреЗрдХреНрд╢рди рдкрд░рддред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/modules/head.py
bias_init()
рдбрд┐рдЯреЗрдХреНрдЯ() рдкреВрд░реНрд╡рд╛рдЧреНрд░рд╣реЛрдВ рдХреЛ рдкреНрд░рд╛рд░рдВрдн рдХрд░реЗрдВ, рдЪреЗрддрд╛рд╡рдиреА: рд╕реНрдЯреНрд░рд╛рдЗрдб рдЙрдкрд▓рдмреНрдзрддрд╛ рдХреА рдЖрд╡рд╢реНрдпрдХрддрд╛ рд╣реИред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/modules/head.py
forward(x, text)
рдЕрдиреБрдорд╛рдирд┐рдд рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдФрд░ рд╡рд░реНрдЧ рд╕рдВрднрд╛рд╡рдирд╛рдУрдВ рдХреЛ рдЬреЛрдбрд╝рддрд╛ рд╣реИ рдФрд░ рд▓реМрдЯрд╛рддрд╛ рд╣реИред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/modules/head.py
ultralytics.nn.modules.head.RTDETRDecoder
рдХрд╛ рд░реВрдк: Module
рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди рдХреЗ рд▓рд┐рдП рд░реАрдпрд▓-рдЯрд╛рдЗрдо рдбрд┐рдлреЙрд░реНрдореЗрдмрд▓ рдЯреНрд░рд╛рдВрд╕рдлрд╛рд░реНрдорд░ рдбрд┐рдХреЛрдбрд░ (RTDETRDecoder) рдореЙрдбреНрдпреВрд▓ред
рдпрд╣ рдбрд┐рдХреЛрдбрд░ рдореЙрдбреНрдпреВрд▓ рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдХреА рднрд╡рд┐рд╖реНрдпрд╡рд╛рдгреА рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рд╡рд┐рдХреГрдд convolutions рдХреЗ рд╕рд╛рде рдЯреНрд░рд╛рдВрд╕рдлрд╛рд░реНрдорд░ рдЖрд░реНрдХрд┐рдЯреЗрдХреНрдЪрд░ рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░рддрд╛ рд╣реИ рдФрд░ рдПрдХ рдЫрд╡рд┐ рдореЗрдВ рд╡рд╕реНрддреБрдУрдВ рдХреЗ рд▓рд┐рдП рд╡рд░реНрдЧ рд▓реЗрдмрд▓ред рдпрд╣ рдХрдИ рдкрд░рддреЛрдВ рд╕реЗ рд╕реБрд╡рд┐рдзрд╛рдУрдВ рдХреЛ рдПрдХреАрдХреГрдд рдХрд░рддрд╛ рд╣реИ рдФрд░ рдХреА рдПрдХ рд╢реНрд░реГрдВрдЦрд▓рд╛ рдХреЗ рдорд╛рдзреНрдпрдо рд╕реЗ рдЪрд▓рддрд╛ рд╣реИ рдЕрдВрддрд┐рдо рднрд╡рд┐рд╖реНрдпрд╡рд╛рдгрд┐рдпреЛрдВ рдХреЛ рдЖрдЙрдЯрдкреБрдЯ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдЯреНрд░рд╛рдВрд╕рдлрд╛рд░реНрдорд░ рдбрд┐рдХреЛрдбрд░ рдкрд░рддреЗрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/modules/head.py
261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 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 |
|
__init__(nc=80, ch=(512, 1024, 2048), hd=256, nq=300, ndp=4, nh=8, ndl=6, d_ffn=1024, dropout=0.0, act=nn.ReLU(), eval_idx=-1, nd=100, label_noise_ratio=0.5, box_noise_scale=1.0, learnt_init_query=False)
рджрд┐рдП рдЧрдП рдорд╛рдкрджрдВрдбреЛрдВ рдХреЗ рд╕рд╛рде RTDETRDecoder рдореЙрдбреНрдпреВрд▓ рдХреЛ рдЗрдирд┐рд╢рд┐рдпрд▓рд╛рдЗрдЬрд╝ рдХрд░рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
nc |
int
|
рдХрдХреНрд╖рд╛рдУрдВ рдХреА рд╕рдВрдЦреНрдпрд╛ред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 80 рд╣реИред |
80
|
ch |
tuple
|
рдмреИрдХрдмреЛрди рдореЗрдВ рдЪреИрдирд▓ рдирдХреНрд╢реЗ рдкреЗрд╢ рдХрд░рддреЗ рд╣реИрдВред рдбрд┐рдлрд╝реЙрд▓реНрдЯ (512, 1024, 2048) рд╣реИред |
(512, 1024, 2048)
|
hd |
int
|
рдЫрд┐рдкреА рд╣реБрдИ рдкрд░рддреЛрдВ рдХрд╛ рдЖрдпрд╛рдоред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 256 рд╣реИред |
256
|
nq |
int
|
рдХреНрд╡реЗрд░реА рдмрд┐рдВрджреБрдУрдВ рдХреА рд╕рдВрдЦреНрдпрд╛. рдбрд┐рдлрд╝реЙрд▓реНрдЯ 300 рд╣реИред |
300
|
ndp |
int
|
рдбрд┐рдХреЛрдбрд░ рдмрд┐рдВрджреБрдУрдВ рдХреА рд╕рдВрдЦреНрдпрд╛ред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 4 рд╣реИред |
4
|
nh |
int
|
рдмрд╣реБ-рд╕рд┐рд░ рдзреНрдпрд╛рди рдореЗрдВ рд╕рд┐рд░ рдХреА рд╕рдВрдЦреНрдпрд╛ред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 8 рд╣реИред |
8
|
ndl |
int
|
рдбрд┐рдХреЛрдбрд░ рдкрд░рддреЛрдВ рдХреА рд╕рдВрдЦреНрдпрд╛ред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 6 рд╣реИред |
6
|
d_ffn |
int
|
рдлрд╝реАрдб-рдлрд╝реЙрд░рд╡рд░реНрдб рдиреЗрдЯрд╡рд░реНрдХ рдХрд╛ рдЖрдпрд╛рдо. рдбрд┐рдлрд╝реЙрд▓реНрдЯ 1024 рд╣реИред |
1024
|
dropout |
float
|
рдбреНрд░реЙрдкрдЖрдЙрдЯ рджрд░ред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 0 рд╣реИред |
0.0
|
act |
Module
|
рд╕рдХреНрд░рд┐рдпрдг рд╕рдорд╛рд░реЛрд╣ред рдбрд┐рдлрд╝реЙрд▓реНрдЯ nn рд╣реИ. рд░реЗрд▓реВред |
ReLU()
|
eval_idx |
int
|
рдореВрд▓реНрдпрд╛рдВрдХрди рд╕реВрдЪрдХрд╛рдВрдХред рдбрд┐рдлрд╝реЙрд▓реНрдЯ -1 рд╣реИред |
-1
|
nd |
int
|
рдЗрдирдХрд╛рд░ рдХреА рд╕рдВрдЦреНрдпрд╛ред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 100 рд╣реИред |
100
|
label_noise_ratio |
float
|
рд▓реЗрдмрд▓ рд╢реЛрд░ рдЕрдиреБрдкрд╛рддред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 0.5 рд╣реИред |
0.5
|
box_noise_scale |
float
|
рдмреЙрдХреНрд╕ рд╢реЛрд░ рдкреИрдорд╛рдиреЗред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 1.0 рд╣реИред |
1.0
|
learnt_init_query |
bool
|
рдкреНрд░рд╛рд░рдВрднрд┐рдХ рдХреНрд╡реЗрд░реА рдПрдореНрдмреЗрдбрд┐рдВрдЧ рд╕реАрдЦрдирд╛ рд╣реИ рдпрд╛ рдирд╣реАрдВред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рдЧрд╝рд▓рдд рд╣реИ. |
False
|
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/modules/head.py
272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 |
|
forward(x, batch=None)
рдореЙрдбреНрдпреВрд▓ рдХреЗ рдЖрдЧреЗ рдкрд╛рд╕ рдЪрд▓рд╛рддрд╛ рд╣реИ, рдЗрдирдкреБрдЯ рдХреЗ рд▓рд┐рдП рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдФрд░ рд╡рд░реНрдЧреАрдХрд░рдг рд╕реНрдХреЛрд░ рд▓реМрдЯрд╛рддрд╛ рд╣реИред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/nn/modules/head.py
2023-11-12 рдмрдирд╛рдпрд╛ рдЧрдпрд╛, рдЕрдкрдбреЗрдЯ рдХрд┐рдпрд╛ рдЧрдпрд╛ 2024-03-03
рд▓реЗрдЦрдХ: рдЧреНрд▓реЗрди-рдЬреЛрдЪрд░ (5)