рдХреЗ рд▓рд┐рдП рд╕рдВрджрд░реНрдн ultralytics/utils/metrics.py
рдиреЛрдЯ
рдпрд╣ рдлрд╝рд╛рдЗрд▓ рдпрд╣рд╛рдБ рдЙрдкрд▓рдмреНрдз рд╣реИ https://github.com/ultralytics/ultralytics/рдмреВрдБрдж/рдореБрдЦреНрдп/ultralytics/utils/metrics.py рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░реЗрдВред рдпрджрд┐ рдЖрдк рдХреЛрдИ рд╕рдорд╕реНрдпрд╛ рджреЗрдЦрддреЗ рд╣реИрдВ рддреЛ рдХреГрдкрдпрд╛ рдкреБрд▓ рдЕрдиреБрд░реЛрдз рдХрд╛ рдпреЛрдЧрджрд╛рди рдХрд░рдХреЗ рдЗрд╕реЗ рдареАрдХ рдХрд░рдиреЗ рдореЗрдВ рдорджрдж рдХрд░реЗрдВ ЁЯЫая╕Пред ЁЯЩП рдзрдиреНрдпрд╡рд╛рдж !
ultralytics.utils.metrics.ConfusionMatrix
рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди рдФрд░ рд╡рд░реНрдЧреАрдХрд░рдг рдХрд╛рд░реНрдпреЛрдВ рдХреЗ рд▓рд┐рдП рднреНрд░рдо рдореИрдЯреНрд░рд┐рдХреНрд╕ рдХреА рдЧрдгрдирд╛ рдФрд░ рдЕрджреНрдпрддрди рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ рд╡рд░реНрдЧред
рд╡рд┐рд╢реЗрд╖рддрд╛рдПрдБ:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|---|
task |
str
|
рдХрд╛рд░реНрдп рдХрд╛ рдкреНрд░рдХрд╛рд░, рдпрд╛ рддреЛ 'рдкрддрд╛ рд▓рдЧрд╛рдирд╛' рдпрд╛ 'рд╡рд░реНрдЧреАрдХреГрдд'ред |
matrix |
ndarray
|
рднреНрд░рдо рдореИрдЯреНрд░рд┐рдХреНрд╕, рдХрд╛рд░реНрдп рдХреЗ рдЖрдзрд╛рд░ рдкрд░ рдЖрдпрд╛рдореЛрдВ рдХреЗ рд╕рд╛рдеред |
nc |
int
|
рдХрдХреНрд╖рд╛рдУрдВ рдХреА рд╕рдВрдЦреНрдпрд╛. |
conf |
float
|
рдкрддрд╛ рд▓рдЧрд╛рдиреЗ рдХреЗ рд▓рд┐рдП рдЖрддреНрдорд╡рд┐рд╢реНрд╡рд╛рд╕ рджрд╣рд▓реАрдЬред |
iou_thres |
float
|
рд╕рдВрдШ рдХреА рджрд╣рд▓реАрдЬ рдкрд░ рдЪреМрд░рд╛рд╣рд╛ред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
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 |
|
__init__(nc, conf=0.25, iou_thres=0.45, task='detect')
рдХреЗ рд▓рд┐рдП рд╡рд┐рд╢реЗрд╖рддрд╛рдПрдБ рдкреНрд░рд╛рд░рдВрдн рдХрд░реЗрдВ YOLO рдХреЛ рдЧрдврд╝рдирд╛ред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
matrix()
plot(normalize=True, save_dir='', names=(), on_plot=None)
рд╕реАрдмреЙрд░реНрди рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░рдХреЗ рднреНрд░рдо рдореИрдЯреНрд░рд┐рдХреНрд╕ рдХреЛ рдкреНрд▓реЙрдЯ рдХрд░реЗрдВ рдФрд░ рдЗрд╕реЗ рдПрдХ рдлрд╝рд╛рдЗрд▓ рдореЗрдВ рд╕рд╣реЗрдЬреЗрдВред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
normalize |
bool
|
рднреНрд░рдо рдореИрдЯреНрд░рд┐рдХреНрд╕ рдХреЛ рд╕рд╛рдорд╛рдиреНрдп рдХрд░рдирд╛ рд╣реИ рдпрд╛ рдирд╣реАрдВред |
True
|
save_dir |
str
|
рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рдЬрд╣рд╛рдВ рдкреНрд▓реЙрдЯ рд╕рд╣реЗрдЬрд╛ рдЬрд╛рдПрдЧрд╛ред |
''
|
names |
tuple
|
рднреВрдЦрдВрдб рдкрд░ рд▓реЗрдмрд▓ рдХреЗ рд░реВрдк рдореЗрдВ рдЙрдкрдпреЛрдЧ рдХрд┐рдП рдЬрд╛рдиреЗ рд╡рд╛рд▓реЗ рд╡рд░реНрдЧреЛрдВ рдХреЗ рдирд╛рдоред |
()
|
on_plot |
func
|
рдкреНрд▓реЙрдЯреНрд╕ рдкрде рдФрд░ рдбреЗрдЯрд╛ рдХреЛ рдкрд╛рд╕ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ рд╡реИрдХрд▓реНрдкрд┐рдХ рдХреЙрд▓рдмреИрдХ рдЬрдм рд╡реЗ рдкреНрд░рд╕реНрддреБрдд рдХрд┐рдП рдЬрд╛рддреЗ рд╣реИрдВред |
None
|
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
print()
рдХрдВрд╕реЛрд▓ рдкрд░ рднреНрд░рдо рдореИрдЯреНрд░рд┐рдХреНрд╕ рдкреНрд░рд┐рдВрдЯ рдХрд░реЗрдВред
process_batch(detections, gt_bboxes, gt_cls)
рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди рдХрд╛рд░реНрдп рдХреЗ рд▓рд┐рдП рднреНрд░рдо рдореИрдЯреНрд░рд┐рдХреНрд╕ рдЕрдкрдбреЗрдЯ рдХрд░реЗрдВред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
detections |
Array[N, 6] | Array[N, 7]
|
рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдФрд░ рдЙрдирдХреА рд╕рдВрдмрдВрдзрд┐рдд рдЬрд╛рдирдХрд╛рд░реА рдХрд╛ рдкрддрд╛ рд▓рдЧрд╛рдпрд╛ред
рдкреНрд░рддреНрдпреЗрдХ рдкрдВрдХреНрддрд┐ рдореЗрдВ (x1, y1, x2, y2, conf, class) рд╣реЛрдирд╛ рдЪрд╛рд╣рд┐рдП
рдпрд╛ рдПрдХ рдЕрддрд┐рд░рд┐рдХреНрдд рддрддреНрд╡ рдХреЗ рд╕рд╛рде |
рдЖрд╡рд╢реНрдпрдХ |
gt_bboxes |
Array[M, 4] | Array[N, 5]
|
xyxy/xyxyr рдкреНрд░рд╛рд░реВрдк рдХреЗ рд╕рд╛рде рдЧреНрд░рд╛рдЙрдВрдб рдЯреНрд░реБрде рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ред |
рдЖрд╡рд╢реНрдпрдХ |
gt_cls |
Array[M]
|
рд╡рд░реНрдЧ рд▓реЗрдмрд▓ред |
рдЖрд╡рд╢реНрдпрдХ |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
process_cls_preds(preds, targets)
рд╡рд░реНрдЧреАрдХрд░рдг рдХрд╛рд░реНрдп рдХреЗ рд▓рд┐рдП рднреНрд░рдо рдореИрдЯреНрд░рд┐рдХреНрд╕ рдЕрдкрдбреЗрдЯ рдХрд░реЗрдВред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
preds |
Array[N, min(nc, 5)]
|
рдЕрдиреБрдорд╛рдирд┐рдд рд╡рд░реНрдЧ рд▓реЗрдмрд▓ред |
рдЖрд╡рд╢реНрдпрдХ |
targets |
Array[N, 1]
|
рдЧреНрд░рд╛рдЙрдВрдб рдЯреНрд░реБрде рдХреНрд▓рд╛рд╕ рд▓реЗрдмрд▓ред |
рдЖрд╡рд╢реНрдпрдХ |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
tp_fp()
рд╕рдЪреНрдЪреА рд╕рдХрд╛рд░рд╛рддреНрдордХрддрд╛ рдФрд░ рдЭреВрдареА рд╕рдХрд╛рд░рд╛рддреНрдордХрддрд╛ рд▓реМрдЯрд╛рддрд╛ рд╣реИред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
ultralytics.utils.metrics.Metric
рдХрд╛ рд░реВрдк: SimpleClass
рдХреЗ рд▓рд┐рдП рдореВрд▓реНрдпрд╛рдВрдХрди рдореЗрдЯреНрд░рд┐рдХреНрд╕ рдХреА рдЧрдгрдирд╛ рдХреЗ рд▓рд┐рдП рд╡рд░реНрдЧ YOLOv8 рдХреЛ рдЧрдврд╝рдирд╛ред
рд╡рд┐рд╢реЗрд╖рддрд╛рдПрдБ:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|---|
p |
list
|
рдкреНрд░рддреНрдпреЗрдХ рд╡рд░реНрдЧ рдХреЗ рд▓рд┐рдП рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ред рдЖрдХрд╛рд░: (рдПрдирд╕реА,)ред |
r |
list
|
рдкреНрд░рддреНрдпреЗрдХ рд╡рд░реНрдЧ рдХреЗ рд▓рд┐рдП рдпрд╛рдж рдХрд░реЗрдВред рдЖрдХрд╛рд░: (рдПрдирд╕реА,)ред |
f1 |
list
|
рдкреНрд░рддреНрдпреЗрдХ рд╡рд░реНрдЧ рдХреЗ рд▓рд┐рдП F1 рд╕реНрдХреЛрд░ред рдЖрдХрд╛рд░: (рдПрдирд╕реА,)ред |
all_ap |
list
|
рд╕рднреА рд╡рд░реНрдЧреЛрдВ рдФрд░ рд╕рднреА IoU рдереНрд░реЗрд╕рд╣реЛрд▓реНрдб рдХреЗ рд▓рд┐рдП AP рд╕реНрдХреЛрд░ред рдЖрдХрд╛рд░: (рдПрдирд╕реА, 10)ред |
ap_class_index |
list
|
рдкреНрд░рддреНрдпреЗрдХ рдПрдкреА рд╕реНрдХреЛрд░ рдХреЗ рд▓рд┐рдП рдХрдХреНрд╖рд╛ рдХрд╛ рд╕реВрдЪрдХрд╛рдВрдХред рдЖрдХрд╛рд░: (рдПрдирд╕реА,)ред |
nc |
int
|
рдХрдХреНрд╖рд╛рдУрдВ рдХреА рд╕рдВрдЦреНрдпрд╛ред |
рд╡рд┐рдзрд┐рдпрд╛рдБ:
рдирд╛рдо | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
ap50 |
рд╕рднреА рд╡рд░реНрдЧреЛрдВ рдХреЗ рд▓рд┐рдП 0.5 рдХреА IoU рд╕реАрдорд╛ рдкрд░ APред рд░рд┐рдЯрд░реНрди: рдПрдкреА рд╕реНрдХреЛрд░ рдХреА рд╕реВрдЪреАред рдЖрдХрд╛рд░: (рдПрдирд╕реА,) рдпрд╛ []ред |
ap |
рд╕рднреА рд╡рд░реНрдЧреЛрдВ рдХреЗ рд▓рд┐рдП 0.5 рд╕реЗ 0.95 рддрдХ IoU рдереНрд░реЗрд╕рд╣реЛрд▓реНрдб рдкрд░ APред рд░рд┐рдЯрд░реНрди: рдПрдкреА рд╕реНрдХреЛрд░ рдХреА рд╕реВрдЪреАред рдЖрдХрд╛рд░: (рдПрдирд╕реА,) рдпрд╛ []ред |
mp |
рд╕рднреА рд╡рд░реНрдЧреЛрдВ рдХреА рдФрд╕рдд рд╕рдЯреАрдХрддрд╛ред рд░рд┐рдЯрд░реНрди: рдлреНрд▓реЛрдЯред |
mr |
рд╕рднреА рд╡рд░реНрдЧреЛрдВ рдХрд╛ рдорддрд▓рдм рдпрд╛рдж рд╣реИред рд░рд┐рдЯрд░реНрди: рдлреНрд▓реЛрдЯред |
map50 |
рд╕рднреА рд╡рд░реНрдЧреЛрдВ рдХреЗ рд▓рд┐рдП 0.5 рдХреА IoU рд╕реАрдорд╛ рдкрд░ рдФрд╕рдд APред рд░рд┐рдЯрд░реНрди: рдлреНрд▓реЛрдЯред |
map75 |
рд╕рднреА рд╡рд░реНрдЧреЛрдВ рдХреЗ рд▓рд┐рдП 0.75 рдХреА IoU рд╕реАрдорд╛ рдкрд░ рдФрд╕рдд APред рд░рд┐рдЯрд░реНрди: рдлреНрд▓реЛрдЯред |
map |
рд╕рднреА рд╡рд░реНрдЧреЛрдВ рдХреЗ рд▓рд┐рдП 0.5 рд╕реЗ 0.95 рддрдХ IoU рдереНрд░реЗрд╕рд╣реЛрд▓реНрдб рдкрд░ рдФрд╕рдд APред рд░рд┐рдЯрд░реНрди: рдлреНрд▓реЛрдЯред |
mean_results |
рдкрд░рд┐рдгрд╛рдореЛрдВ рдХрд╛ рдорддрд▓рдм, рд░рд┐рдЯрд░реНрди рдПрдордкреА, рдПрдордЖрд░, рдореИрдк50, рдореИрдкред |
class_result |
рдХреНрд▓рд╛рд╕-рдЕрд╡реЗрдпрд░ рдкрд░рд┐рдгрд╛рдо, p[i], r[i], ap50[i], ap[i] рд▓реМрдЯрд╛рддрд╛ рд╣реИред |
maps |
рдкреНрд░рддреНрдпреЗрдХ рд╡рд░реНрдЧ рдХрд╛ рдПрдордПрдкреАред рд░рд┐рдЯрд░реНрди: рдПрдордПрдкреА рд╕реНрдХреЛрд░ рдХреА рд╕рд░рдгреА, рдЖрдХрд╛рд░: (рдПрдирд╕реА,)ред |
fitness |
рдореИрдЯреНрд░рд┐рдХреНрд╕ рдХреЗ рднрд╛рд░рд┐рдд рд╕рдВрдпреЛрдЬрди рдХреЗ рд░реВрдк рдореЗрдВ рдореЙрдбрд▓ рдлрд┐рдЯрдиреЗрд╕ред рд░рд┐рдЯрд░реНрди: рдлреНрд▓реЛрдЯред |
update |
рдирдП рдореВрд▓реНрдпрд╛рдВрдХрди рдкрд░рд┐рдгрд╛рдореЛрдВ рдХреЗ рд╕рд╛рде рдореАрдЯреНрд░рд┐рдХ рд╡рд┐рд╢реЗрд╖рддрд╛рдПрдБ рдЕрдкрдбреЗрдЯ рдХрд░реЗрдВ. |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
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 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 |
|
ap
property
рд╕рднреА рд╡рд░реНрдЧреЛрдВ рдХреЗ рд▓рд┐рдП 0.5-0.95 рдХреА IoU рд╕реАрдорд╛ рдкрд░ рдФрд╕рдд рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ (AP) рджреЗрддрд╛ рд╣реИ.
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
(ndarray, list)
|
рдкреНрд░рддрд┐ рд╡рд░реНрдЧ AP50-95 рдорд╛рдиреЛрдВ рдХреЗ рд╕рд╛рде рдЖрдХреГрддрд┐ (nc,) рдХреА рд╕рд░рдгреА, рдпрд╛ рдпрджрд┐ рдЙрдкрд▓рдмреНрдз рдирд╣реАрдВ рд╣реИ рддреЛ рдПрдХ рдЦрд╛рд▓реА рд╕реВрдЪреАред |
ap50
property
рд╕рднреА рд╡рд░реНрдЧреЛрдВ рдХреЗ рд▓рд┐рдП 0.5 рдХреА IoU рд╕реАрдорд╛ рдкрд░ рдФрд╕рдд рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ (AP) рджреЗрддрд╛ рд╣реИ.
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
(ndarray, list)
|
рдкреНрд░рддрд┐ рд╡рд░реНрдЧ AP50 рдорд╛рдиреЛрдВ рдХреЗ рд╕рд╛рде рдЖрдХреГрддрд┐ (nc,) рдХреА рд╕рд░рдгреА, рдпрд╛ рдпрджрд┐ рдЙрдкрд▓рдмреНрдз рдирд╣реАрдВ рд╣реИ рддреЛ рдПрдХ рдЦрд╛рд▓реА рд╕реВрдЪреАред |
curves
property
рд╡рд┐рд╢рд┐рд╖реНрдЯ рдореАрдЯреНрд░рд┐рдХ рд╡рдХреНрд░реЛрдВ рддрдХ рдкрд╣реБрдБрдЪрдиреЗ рдХреЗ рд▓рд┐рдП рд╡рдХреНрд░реЛрдВ рдХреА рдПрдХ рд╕реВрдЪреА рджреЗрддрд╛ рд╣реИред
curves_results
property
рд╡рд┐рд╢рд┐рд╖реНрдЯ рдореАрдЯреНрд░рд┐рдХ рд╡рдХреНрд░реЛрдВ рддрдХ рдкрд╣реБрдБрдЪрдиреЗ рдХреЗ рд▓рд┐рдП рд╡рдХреНрд░реЛрдВ рдХреА рдПрдХ рд╕реВрдЪреА рджреЗрддрд╛ рд╣реИред
map
property
0.05 рдХреЗ рдЪрд░рдгреЛрдВ рдореЗрдВ 0.5 - 0.95 рдХреА IoU рдереНрд░реЗрд╢реЛрд▓реНрдб рдкрд░ рдорд╛рдзреНрдп рдФрд╕рдд рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ (mAP) рджреЗрддрд╛ рд╣реИ.
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
float
|
0.05 рдХреЗ рдЪрд░рдгреЛрдВ рдореЗрдВ 0.5 - 0.95 рдХреЗ IoU рдереНрд░реЗрд╕рд╣реЛрд▓реНрдб рдкрд░ mAPред |
map50
property
рдорд╛рдзреНрдп рдФрд╕рдд рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ (mAP) рдХреЛ 0.5 рдХреА IoU рд╕реАрдорд╛ рдкрд░ рджреЗрддрд╛ рд╣реИ.
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
float
|
0.5 рдХреА IoU рд╕реАрдорд╛ рдкрд░ mAPред |
map75
property
рдорд╛рдзреНрдп рдФрд╕рдд рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ (mAP) 0.75 рдХреА IoU рд╕реАрдорд╛ рдкрд░ рджреЗрддрд╛ рд╣реИ.
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
float
|
0.75 рдХреА IoU рд╕реАрдорд╛ рдкрд░ mAPред |
maps
property
рдкреНрд░рддреНрдпреЗрдХ рд╡рд░реНрдЧ рдХрд╛ рдирдХреНрд╢рд╛ред
mp
property
рд╕рднреА рд╡рд░реНрдЧреЛрдВ рдХреА рдорд╛рдзреНрдп рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ рджреЗрддрд╛ рд╣реИ.
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
float
|
рд╕рднреА рд╡рд░реНрдЧреЛрдВ рдХреА рдФрд╕рдд рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ред |
mr
property
рд╕рднреА рд╡рд░реНрдЧреЛрдВ рдХрд╛ рдореАрди рд░рд┐рдХреЙрд▓ рджреЗрддрд╛ рд╣реИред
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
float
|
рд╕рднреА рд╡рд░реНрдЧреЛрдВ рдХрд╛ рдорд╛рдзреНрдп рд╕реНрдорд░рдгред |
__init__()
рдХреЗ рд▓рд┐рдП рдореВрд▓реНрдпрд╛рдВрдХрди рдореЗрдЯреНрд░рд┐рдХреНрд╕ рдХреА рдЧрдгрдирд╛ рдХреЗ рд▓рд┐рдП рдПрдХ рдореАрдЯреНрд░рд┐рдХ рдЗрдВрд╕реНрдЯреЗрдВрд╕ рдХреЛ рдЗрдирд┐рд╢рд┐рдпрд▓рд╛рдЗрдЬрд╝ рдХрд░рддрд╛ рд╣реИ YOLOv8 рдХреЛ рдЧрдврд╝рдирд╛ред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
class_result(i)
рд╡рд░реНрдЧ-рдЬрд╛рдЧрд░реВрдХ рдкрд░рд┐рдгрд╛рдо, рд░рд┐рдЯрд░реНрди рдкреА [рдЖрдИ], рдЖрд░ [рдЖрдИ], рдПрдкреА 50 [рдЖрдИ], рдПрдкреА [рдЖрдИ]ред
fitness()
рдореИрдЯреНрд░рд┐рдХреНрд╕ рдХреЗ рднрд╛рд░рд┐рдд рд╕рдВрдпреЛрдЬрди рдХреЗ рд░реВрдк рдореЗрдВ рдореЙрдбрд▓ рдлрд┐рдЯрдиреЗрд╕ред
mean_results()
рдкрд░рд┐рдгрд╛рдореЛрдВ рдХрд╛ рдорддрд▓рдм, рд╡рд╛рдкрд╕реА рдПрдордкреА, рд╢реНрд░реА, рдорд╛рдирдЪрд┐рддреНрд░ 50, рдирдХреНрд╢рд╛ред
update(results)
рдкрд░рд┐рдгрд╛рдореЛрдВ рдХреЗ рдПрдХ рдирдП рд╕реЗрдЯ рдХреЗ рд╕рд╛рде рдореЙрдбрд▓ рдХреЗ рдореВрд▓реНрдпрд╛рдВрдХрди рдореАрдЯреНрд░рд┐рдХ рдХреЛ рдЕрдкрдбреЗрдЯ рдХрд░рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
results |
tuple
|
рдПрдХ рдЯрдкрд▓ рдЬрд┐рд╕рдореЗрдВ рдирд┐рдореНрдирд▓рд┐рдЦрд┐рдд рдореВрд▓реНрдпрд╛рдВрдХрди рдореАрдЯреНрд░рд┐рдХ рд╢рд╛рдорд┐рд▓ рд╣реИрдВ: - рдкреА (рд╕реВрдЪреА): рдкреНрд░рддреНрдпреЗрдХ рд╡рд░реНрдЧ рдХреЗ рд▓рд┐рдП рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ред рдЖрдХрд╛рд░: (рдПрдирд╕реА,)ред - r (рд╕реВрдЪреА): рдкреНрд░рддреНрдпреЗрдХ рд╡рд░реНрдЧ рдХреЗ рд▓рд┐рдП рдпрд╛рдж рдХрд░реЗрдВред рдЖрдХрд╛рд░: (рдПрдирд╕реА,)ред - f1 (рд╕реВрдЪреА): рдкреНрд░рддреНрдпреЗрдХ рд╡рд░реНрдЧ рдХреЗ рд▓рд┐рдП F1 рд╕реНрдХреЛрд░ред рдЖрдХрд╛рд░: (рдПрдирд╕реА,)ред - all_ap (рд╕реВрдЪреА): рд╕рднреА рд╡рд░реНрдЧреЛрдВ рдФрд░ рд╕рднреА IoU рдереНрд░реЗрд╕рд╣реЛрд▓реНрдб рдХреЗ рд▓рд┐рдП AP рд╕реНрдХреЛрд░ред рдЖрдХрд╛рд░: (рдПрдирд╕реА, 10)ред - ap_class_index (рд╕реВрдЪреА): рдкреНрд░рддреНрдпреЗрдХ рдПрдкреА рд╕реНрдХреЛрд░ рдХреЗ рд▓рд┐рдП рдХрдХреНрд╖рд╛ рдХрд╛ рд╕реВрдЪрдХрд╛рдВрдХред рдЖрдХрд╛рд░: (рдПрдирд╕реА,)ред |
рдЖрд╡рд╢реНрдпрдХ |
рджреБрд╖реНрдкреНрд░рднрд╛рд╡
рд╡рд░реНрдЧ рд╡рд┐рд╢реЗрд╖рддрд╛рдУрдВ рдХреЛ рдЕрджреНрдпрддрди рдХрд░рддрд╛ рд╣реИ self.p
, self.r
, self.f1
, self.all_ap
рдФрд░ self.ap_class_index
рдЖрдзрд╛рд░рд┐рдд
рдореЗрдВ рдкреНрд░рджрд╛рди рдХрд┐рдП рдЧрдП рдореВрд▓реНрдпреЛрдВ рдкрд░ results
рдЯрдкрд▓ред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
ultralytics.utils.metrics.DetMetrics
рдХрд╛ рд░реВрдк: SimpleClass
рдпрд╣ рд╡рд░реНрдЧ рд╕рдЯреАрдХ, рдпрд╛рдж рдФрд░ рдФрд╕рдд рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ рдЬреИрд╕реЗ рдбрд┐рдЯреЗрдХреНрд╢рди рдореЗрдЯреНрд░рд┐рдХреНрд╕ рдХреА рдЧрдгрдирд╛ рдХреЗ рд▓рд┐рдП рдПрдХ рдЙрдкрдпреЛрдЧрд┐рддрд╛ рд╡рд░реНрдЧ рд╣реИ (рдПрдордПрдкреА) рдПрдХ рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди рдореЙрдбрд▓ рдХрд╛ред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
save_dir |
Path
|
рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рдХреЗ рд▓рд┐рдП рдПрдХ рдкрде рдЬрд╣рд╛рдВ рдЖрдЙрдЯрдкреБрдЯ рдкреНрд▓реЙрдЯ рд╕рд╣реЗрдЬреЗ рдЬрд╛рдПрдВрдЧреЗред рд╡рд░реНрддрдорд╛рди рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
Path('.')
|
plot |
bool
|
рдПрдХ рдзреНрд╡рдЬ рдЬреЛ рдЗрдВрдЧрд┐рдд рдХрд░рддрд╛ рд╣реИ рдХрд┐ рдкреНрд░рддреНрдпреЗрдХ рд╡рд░реНрдЧ рдХреЗ рд▓рд┐рдП рд╕рдЯреАрдХ-рдпрд╛рдж рдШрдЯрддрд╛ рд╣реИ рдпрд╛ рдирд╣реАрдВред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рд░реВрдк рд╕реЗ рдЧрд▓рдд рд╣реИ. |
False
|
on_plot |
func
|
рдкреНрд▓реЙрдЯреНрд╕ рдкрде рдФрд░ рдбреЗрдЯрд╛ рдХреЛ рдкрд╛рд╕ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ рд╡реИрдХрд▓реНрдкрд┐рдХ рдХреЙрд▓рдмреИрдХ рдЬрдм рд╡реЗ рдкреНрд░рд╕реНрддреБрдд рдХрд┐рдП рдЬрд╛рддреЗ рд╣реИрдВред рдХреЛрдИ рдирд╣реАрдВ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
None
|
names |
tuple of str
|
рддрд╛рд░реЛрдВ рдХрд╛ рдПрдХ рдЯрдкрд▓ рдЬреЛ рд╡рд░реНрдЧреЛрдВ рдХреЗ рдирд╛рдореЛрдВ рдХрд╛ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ рдХрд░рддрд╛ рд╣реИред рдПрдХ рдЦрд╛рд▓реА рдЯрдкрд▓ рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
()
|
рд╡рд┐рд╢реЗрд╖рддрд╛рдПрдБ:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|---|
save_dir |
Path
|
рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рдХреЗ рд▓рд┐рдП рдПрдХ рдкрде рдЬрд╣рд╛рдВ рдЖрдЙрдЯрдкреБрдЯ рдкреНрд▓реЙрдЯ рд╕рд╣реЗрдЬреЗ рдЬрд╛рдПрдВрдЧреЗред |
plot |
bool
|
рдПрдХ рдзреНрд╡рдЬ рдЬреЛ рдЗрдВрдЧрд┐рдд рдХрд░рддрд╛ рд╣реИ рдХрд┐ рдкреНрд░рддреНрдпреЗрдХ рд╡рд░реНрдЧ рдХреЗ рд▓рд┐рдП рд╕рдЯреАрдХ-рдпрд╛рдж рдШрдЯрддрд╛ рдХреЛ рдкреНрд▓реЙрдЯ рдХрд░рдирд╛ рд╣реИ рдпрд╛ рдирд╣реАрдВред |
on_plot |
func
|
рдкреНрд▓реЙрдЯреНрд╕ рдкрде рдФрд░ рдбреЗрдЯрд╛ рдХреЛ рдкрд╛рд╕ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ рд╡реИрдХрд▓реНрдкрд┐рдХ рдХреЙрд▓рдмреИрдХ рдЬрдм рд╡реЗ рдкреНрд░рд╕реНрддреБрдд рдХрд┐рдП рдЬрд╛рддреЗ рд╣реИрдВред |
names |
tuple of str
|
рддрд╛рд░реЛрдВ рдХрд╛ рдПрдХ рдЯрдкрд▓ рдЬреЛ рд╡рд░реНрдЧреЛрдВ рдХреЗ рдирд╛рдореЛрдВ рдХрд╛ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ рдХрд░рддрд╛ рд╣реИред |
box |
Metric
|
рдбрд┐рдЯреЗрдХреНрд╢рди рдореЗрдЯреНрд░рд┐рдХреНрд╕ рдХреЗ рдкрд░рд┐рдгрд╛рдореЛрдВ рдХреЛ рд╕рдВрдЧреНрд░рд╣реАрдд рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдореАрдЯреНрд░рд┐рдХ рд╡рд░реНрдЧ рдХрд╛ рдПрдХ рдЙрджрд╛рд╣рд░рдгред |
speed |
dict
|
рдкрддрд╛ рд▓рдЧрд╛рдиреЗ рдХреА рдкреНрд░рдХреНрд░рд┐рдпрд╛ рдХреЗ рд╡рд┐рднрд┐рдиреНрди рднрд╛рдЧреЛрдВ рдХреЗ рдирд┐рд╖реНрдкрд╛рджрди рд╕рдордп рдХреЗ рднрдВрдбрд╛рд░рдг рдХреЗ рд▓рд┐рдП рдПрдХ рд╢рдмреНрджрдХреЛрд╢ред |
рд╡рд┐рдзрд┐рдпрд╛рдБ:
рдирд╛рдо | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
process |
рдкреВрд░реНрд╡рд╛рдиреБрдорд╛рдиреЛрдВ рдХреЗ рдирд╡реАрдирддрдо рдмреИрдЪ рдХреЗ рд╕рд╛рде рдореАрдЯреНрд░рд┐рдХ рдкрд░рд┐рдгрд╛рдореЛрдВ рдХреЛ рдЕрдкрдбреЗрдЯ рдХрд░рддрд╛ рд╣реИред |
keys |
рдкрд░рд┐рдХрд▓рд┐рдд рдбрд┐рдЯреЗрдХреНрд╢рди рдореЗрдЯреНрд░рд┐рдХреНрд╕ рддрдХ рдкрд╣реБрдБрдЪрдиреЗ рдХреЗ рд▓рд┐рдП рдХреБрдВрдЬрд┐рдпреЛрдВ рдХреА рдПрдХ рд╕реВрдЪреА рджреЗрддрд╛ рд╣реИ. |
mean_results |
рдкрд░рд┐рдХрд▓рд┐рдд рдбрд┐рдЯреЗрдХреНрд╢рди рдореЗрдЯреНрд░рд┐рдХреНрд╕ рдХреЗ рд▓рд┐рдП рдорд╛рдзреНрдп рдорд╛рдиреЛрдВ рдХреА рдПрдХ рд╕реВрдЪреА рджреЗрддрд╛ рд╣реИ. |
class_result |
рдХрд┐рд╕реА рд╡рд┐рд╢рд┐рд╖реНрдЯ рд╡рд░реНрдЧ рдХреЗ рд▓рд┐рдП рдкрд░рд┐рдХрд▓рд┐рдд рдбрд┐рдЯреЗрдХреНрд╢рди рдореЗрдЯреНрд░рд┐рдХреНрд╕ рдХреЗ рдорд╛рдиреЛрдВ рдХреА рд╕реВрдЪреА рджреЗрддрд╛ рд╣реИ. |
maps |
рд╡рд┐рднрд┐рдиреНрди IoU рдереНрд░реЗрд╢реЛрд▓реНрдб рдХреЗ рд▓рд┐рдП рдорд╛рдзреНрдп рдФрд╕рдд рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ (mAP) рдорд╛рдиреЛрдВ рдХрд╛ рд╢рдмреНрджрдХреЛрд╢ рджреЗрддрд╛ рд╣реИ. |
fitness |
рдЧрдгрдирд╛ рдХреА рдЧрдИ рдкрд╣рдЪрд╛рди рдореАрдЯреНрд░рд┐рдХ рдХреЗ рдЖрдзрд╛рд░ рдкрд░ рдлрд┐рдЯрдиреЗрд╕ рд╕реНрдХреЛрд░ рдХреА рдЧрдгрдирд╛ рдХрд░рддрд╛ рд╣реИред |
ap_class_index |
рд╡рд░реНрдЧ рд╕реВрдЪрдХрд╛рдВрдХреЛрдВ рдХреА рдПрдХ рд╕реВрдЪреА рджреЗрддрд╛ рд╣реИ рдЬрд┐рдиреНрд╣реЗрдВ рдЙрдирдХреЗ рдФрд╕рдд рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ (AP) рдорд╛рдиреЛрдВ рджреНрд╡рд╛рд░рд╛ рдХреНрд░рдордмрджреНрдз рдХрд┐рдпрд╛ рдЬрд╛рддрд╛ рд╣реИ. |
results_dict |
рдПрдХ рд╢рдмреНрджрдХреЛрд╢ рд▓реМрдЯрд╛рддрд╛ рд╣реИ рдЬреЛ рдбрд┐рдЯреЗрдХреНрд╢рди рдореАрдЯреНрд░рд┐рдХ рдХреБрдВрдЬрд┐рдпреЛрдВ рдХреЛ рдЙрдирдХреЗ рдкрд░рд┐рдХрд▓рд┐рдд рдорд╛рдиреЛрдВ рдкрд░ рдореИрдк рдХрд░рддрд╛ рд╣реИ. |
curves |
рдЯреВрдбреЛ |
curves_results |
рдЯреВрдбреЛ |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
ap_class_index
property
рдкреНрд░рддрд┐ рд╡рд░реНрдЧ рдФрд╕рдд рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ рд╕реВрдЪрдХрд╛рдВрдХ рджреЗрддрд╛ рд╣реИ.
curves
property
рд╡рд┐рд╢рд┐рд╖реНрдЯ рдореАрдЯреНрд░рд┐рдХ рд╡рдХреНрд░реЛрдВ рддрдХ рдкрд╣реБрдБрдЪрдиреЗ рдХреЗ рд▓рд┐рдП рд╡рдХреНрд░реЛрдВ рдХреА рдПрдХ рд╕реВрдЪреА рджреЗрддрд╛ рд╣реИред
curves_results
property
рдЧрдгрдирд╛ рдХрд┐рдП рдЧрдП рдкреНрд░рджрд░реНрд╢рди рдореИрдЯреНрд░рд┐рдХреНрд╕ рдФрд░ рдЖрдВрдХрдбрд╝реЛрдВ рдХрд╛ рд╢рдмреНрджрдХреЛрд╢ рджреЗрддрд╛ рд╣реИред
fitness
property
рдмреЙрдХреНрд╕ рдСрдмреНрдЬреЗрдХреНрдЯ рдХреА рдлрд┐рдЯрдиреЗрд╕ рд▓реМрдЯрд╛рддрд╛ рд╣реИред
keys
property
рд╡рд┐рд╢рд┐рд╖реНрдЯ рдореАрдЯреНрд░рд┐рдХ рддрдХ рдкрд╣реБрдБрдЪрдиреЗ рдХреЗ рд▓рд┐рдП рдХреБрдВрдЬрд┐рдпреЛрдВ рдХреА рдПрдХ рд╕реВрдЪреА рджреЗрддрд╛ рд╣реИред
maps
property
рд░рд┐рдЯрд░реНрди рдХрд╛ рдорддрд▓рдм рдкреНрд░рддрд┐ рд╡рд░реНрдЧ рдФрд╕рдд рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ (рдПрдордПрдкреА) рд╕реНрдХреЛрд░ рд╣реИред
results_dict
property
рдЧрдгрдирд╛ рдХрд┐рдП рдЧрдП рдкреНрд░рджрд░реНрд╢рди рдореИрдЯреНрд░рд┐рдХреНрд╕ рдФрд░ рдЖрдВрдХрдбрд╝реЛрдВ рдХрд╛ рд╢рдмреНрджрдХреЛрд╢ рджреЗрддрд╛ рд╣реИред
__init__(save_dir=Path('.'), plot=False, on_plot=None, names=())
рдПрдХ рд╕реЗрд╡ рдбрд╛рдпрд░реЗрдХреНрдЯрд░реА, рдкреНрд▓реЙрдЯ рдлреНрд▓реИрдЧ, рдХреЙрд▓рдмреИрдХ рдлрд╝рдВрдХреНрд╢рди рдФрд░ рдХреНрд▓рд╛рд╕ рдирд╛рдореЛрдВ рдХреЗ рд╕рд╛рде рдПрдХ DetMetrics рдЗрдВрд╕реНрдЯреЗрдВрд╕ рдХреЛ рдЗрдирд┐рд╢рд┐рдпрд▓рд╛рдЗрдЬрд╝ рдХрд░реЗрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
class_result(i)
рдПрдХ рд╡рд┐рд╢рд┐рд╖реНрдЯ рд╡рд░реНрдЧ рдкрд░ рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди рдореЙрдбрд▓ рдХреЗ рдкреНрд░рджрд░реНрд╢рди рдХрд╛ рдореВрд▓реНрдпрд╛рдВрдХрди рдХрд░рдиреЗ рдХрд╛ рдкрд░рд┐рдгрд╛рдо рд▓реМрдЯрд╛рдПрдВред
mean_results()
рдкрддрд╛ рдЪрд▓рд╛ рд╡рд╕реНрддреБрдУрдВ рдХрд╛ рдорддрд▓рдм рдХреА рдЧрдгрдирд╛ & рд╡рд╛рдкрд╕реА рдкрд░рд┐рд╢реБрджреНрдзрддрд╛, рдпрд╛рдж, mAP50, рдФрд░ mAP50-95.
process(tp, conf, pred_cls, target_cls)
рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди рдФрд░ рдЕрдкрдбреЗрдЯ рдореЗрдЯреНрд░рд┐рдХреНрд╕ рдХреЗ рд▓рд┐рдП рдЕрдиреБрдорд╛рдирд┐рдд рдкрд░рд┐рдгрд╛рдореЛрдВ рдХреА рдкреНрд░рдХреНрд░рд┐рдпрд╛ рдХрд░реЗрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
ultralytics.utils.metrics.SegmentMetrics
рдХрд╛ рд░реВрдк: SimpleClass
рдХрдХреНрд╖рд╛рдУрдВ рдХреЗ рдХрд┐рд╕реА рджрд┐рдП рдЧрдП рд╕реЗрдЯ рдкрд░ рдкрддрд╛ рд▓рдЧрд╛рдиреЗ рдФрд░ рд╡рд┐рднрд╛рдЬрди рдореИрдЯреНрд░рд┐рдХреНрд╕ рдХреА рдЧрдгрдирд╛ рдФрд░ рд╕рдореБрдЪреНрдЪрдп рдХрд░рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
save_dir |
Path
|
рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рдХреЗ рд▓рд┐рдП рдкрде рдЬрд╣рд╛рдВ рдЖрдЙрдЯрдкреБрдЯ рдкреНрд▓реЙрдЯ рд╕рд╣реЗрдЬреЗ рдЬрд╛рдиреЗ рдЪрд╛рд╣рд┐рдПред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рд╡рд░реНрддрдорд╛рди рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рд╣реИред |
Path('.')
|
plot |
bool
|
рдЪрд╛рд╣реЗ рдкрддрд╛ рд▓рдЧрд╛рдиреЗ рдФрд░ рд╡рд┐рднрд╛рдЬрди рднреВрдЦрдВрдбреЛрдВ рдХреЛ рдмрдЪрд╛рдиреЗ рдХреЗ рд▓рд┐рдПред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рдЧрд╝рд▓рдд рд╣реИ. |
False
|
on_plot |
func
|
рдкреНрд▓реЙрдЯреНрд╕ рдкрде рдФрд░ рдбреЗрдЯрд╛ рдХреЛ рдкрд╛рд╕ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ рд╡реИрдХрд▓реНрдкрд┐рдХ рдХреЙрд▓рдмреИрдХ рдЬрдм рд╡реЗ рдкреНрд░рд╕реНрддреБрдд рдХрд┐рдП рдЬрд╛рддреЗ рд╣реИрдВред рдХреЛрдИ рдирд╣реАрдВ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
None
|
names |
list
|
рд╡рд░реНрдЧ рдХреЗ рдирд╛рдореЛрдВ рдХреА рд╕реВрдЪреАред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рдПрдХ рд░рд┐рдХреНрдд рд╕реВрдЪреА рд╣реИ. |
()
|
рд╡рд┐рд╢реЗрд╖рддрд╛рдПрдБ:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|---|
save_dir |
Path
|
рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рдХреЗ рд▓рд┐рдП рдкрде рдЬрд╣рд╛рдВ рдЖрдЙрдЯрдкреБрдЯ рдкреНрд▓реЙрдЯ рд╕рд╣реЗрдЬреЗ рдЬрд╛рдиреЗ рдЪрд╛рд╣рд┐рдПред |
plot |
bool
|
рдЪрд╛рд╣реЗ рдкрддрд╛ рд▓рдЧрд╛рдиреЗ рдФрд░ рд╡рд┐рднрд╛рдЬрди рднреВрдЦрдВрдбреЛрдВ рдХреЛ рдмрдЪрд╛рдиреЗ рдХреЗ рд▓рд┐рдПред |
on_plot |
func
|
рдкреНрд▓реЙрдЯреНрд╕ рдкрде рдФрд░ рдбреЗрдЯрд╛ рдХреЛ рдкрд╛рд╕ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ рд╡реИрдХрд▓реНрдкрд┐рдХ рдХреЙрд▓рдмреИрдХ рдЬрдм рд╡реЗ рдкреНрд░рд╕реНрддреБрдд рдХрд┐рдП рдЬрд╛рддреЗ рд╣реИрдВред |
names |
list
|
рд╡рд░реНрдЧ рдХреЗ рдирд╛рдореЛрдВ рдХреА рд╕реВрдЪреАред |
box |
Metric
|
рдмреЙрдХреНрд╕ рдбрд┐рдЯреЗрдХреНрд╢рди рдореЗрдЯреНрд░рд┐рдХреНрд╕ рдХреА рдЧрдгрдирд╛ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдореАрдЯреНрд░рд┐рдХ рд╡рд░реНрдЧ рдХрд╛ рдПрдХ рдЙрджрд╛рд╣рд░рдг. |
seg |
Metric
|
рдорд╛рд╕реНрдХ рд╡рд┐рднрд╛рдЬрди рдореАрдЯреНрд░рд┐рдХ рдХреА рдЧрдгрдирд╛ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдореАрдЯреНрд░рд┐рдХ рд╡рд░реНрдЧ рдХрд╛ рдПрдХ рдЙрджрд╛рд╣рд░рдгред |
speed |
dict
|
рдЕрдиреБрдорд╛рди рдХреЗ рд╡рд┐рднрд┐рдиреНрди рдЪрд░рдгреЛрдВ рдореЗрдВ рд▓рд┐рдП рдЧрдП рд╕рдордп рдХреЛ рд╕рдВрдЧреНрд░рд╣реАрдд рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рд╢рдмреНрджрдХреЛрд╢ред |
рд╡рд┐рдзрд┐рдпрд╛рдБ:
рдирд╛рдо | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
process |
рднрд╡рд┐рд╖реНрдпрд╡рд╛рдгрд┐рдпреЛрдВ рдХреЗ рджрд┐рдП рдЧрдП рд╕реЗрдЯ рдкрд░ рдореИрдЯреНрд░рд┐рдХреНрд╕ рд╕рдВрд╕рд╛рдзрд┐рдд рдХрд░рддрд╛ рд╣реИред |
mean_results |
рд╕рднреА рд╡рд░реНрдЧреЛрдВ рдкрд░ рдкрддрд╛ рд▓рдЧрд╛рдиреЗ рдФрд░ рд╡рд┐рднрд╛рдЬрди рдореИрдЯреНрд░рд┐рдХреНрд╕ рдХрд╛ рдорд╛рдзреНрдп рджреЗрддрд╛ рд╣реИред |
class_result |
рд╡рд░реНрдЧ рдХреА рдЦреЛрдЬ рдФрд░ рд╡рд┐рднрд╛рдЬрди рдореИрдЯреНрд░рд┐рдХреНрд╕ рд▓реМрдЯрд╛рддрд╛ рд╣реИ |
maps |
0.50 рд╕реЗ 0.95 рддрдХ рдХреА IoU рдереНрд░реЗрд╢реЛрд▓реНрдб рдХреЗ рд▓рд┐рдП рдорд╛рдзреНрдп рдФрд╕рдд рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ (mAP) рд╕реНрдХреЛрд░ рджреЗрддрд╛ рд╣реИ. |
fitness |
рдлрд╝рд┐рдЯрдиреЗрд╕ рд╕реНрдХреЛрд░ рд▓реМрдЯрд╛рддрд╛ рд╣реИ, рдЬреЛ рдореЗрдЯреНрд░рд┐рдХреНрд╕ рдХрд╛ рдПрдХрд▓ рднрд╛рд░рд┐рдд рд╕рдВрдпреЛрдЬрди рд╣реЛрддрд╛ рд╣реИ. |
ap_class_index |
рдФрд╕рдд рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ (AP) рдХреА рдЧрдгрдирд╛ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдЙрдкрдпреЛрдЧ рдХрд┐рдП рдЬрд╛рдиреЗ рд╡рд╛рд▓реЗ рд╡рд░реНрдЧреЛрдВ рдХреЗ рд╕реВрдЪрдХрд╛рдВрдХреЛрдВ рдХреА рд╕реВрдЪреА рджреЗрддрд╛ рд╣реИ. |
results_dict |
рд╕рднреА рдкрд╣рдЪрд╛рди рдФрд░ рд╡рд┐рднрд╛рдЬрди рдореИрдЯреНрд░рд┐рдХреНрд╕ рдФрд░ рдлрд┐рдЯрдиреЗрд╕ рд╕реНрдХреЛрд░ рдпреБрдХреНрдд рд╢рдмреНрджрдХреЛрд╢ рджреЗрддрд╛ рд╣реИ. |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
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 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 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 |
|
ap_class_index
property
рдмрдХреНрд╕реЗ рдФрд░ рдорд╛рд╕реНрдХ рдХрд╛ рдПрдХ рд╣реА ap_class_index рд╣реИред
curves
property
рд╡рд┐рд╢рд┐рд╖реНрдЯ рдореАрдЯреНрд░рд┐рдХ рд╡рдХреНрд░реЛрдВ рддрдХ рдкрд╣реБрдБрдЪрдиреЗ рдХреЗ рд▓рд┐рдП рд╡рдХреНрд░реЛрдВ рдХреА рдПрдХ рд╕реВрдЪреА рджреЗрддрд╛ рд╣реИред
curves_results
property
рдЧрдгрдирд╛ рдХрд┐рдП рдЧрдП рдкреНрд░рджрд░реНрд╢рди рдореИрдЯреНрд░рд┐рдХреНрд╕ рдФрд░ рдЖрдВрдХрдбрд╝реЛрдВ рдХрд╛ рд╢рдмреНрджрдХреЛрд╢ рджреЗрддрд╛ рд╣реИред
fitness
property
рд╡рд┐рднрд╛рдЬрди рдФрд░ рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдореЙрдбрд▓ рджреЛрдиреЛрдВ рдХреЗ рд▓рд┐рдП рдлрд┐рдЯрдиреЗрд╕ рд╕реНрдХреЛрд░ рдкреНрд░рд╛рдкреНрдд рдХрд░реЗрдВред
keys
property
рдореАрдЯреНрд░рд┐рдХ рддрдХ рдкрд╣реБрдБрдЪрдиреЗ рдХреЗ рд▓рд┐рдП рдХреБрдВрдЬрд┐рдпреЛрдВ рдХреА рдПрдХ рд╕реВрдЪреА рджреЗрддрд╛ рд╣реИред
maps
property
рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди рдФрд░ рд╕рд┐рдореЗрдВрдЯрд┐рдХ рд╕реЗрдЧрдореЗрдВрдЯреЗрд╢рди рдореЙрдбрд▓ рдХреЗ рд▓рд┐рдП mAP рд╕реНрдХреЛрд░ рджреЗрддрд╛ рд╣реИред
results_dict
property
рдореВрд▓реНрдпрд╛рдВрдХрди рдХреЗ рд▓рд┐рдП рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди рдореЙрдбрд▓ рдХреЗ рдкрд░рд┐рдгрд╛рдо рджреЗрддрд╛ рд╣реИред
__init__(save_dir=Path('.'), plot=False, on_plot=None, names=())
рдПрдХ рд╕рд╣реЗрдЬреЗрдВ рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛, рдкреНрд▓реЙрдЯ рдзреНрд╡рдЬ, рдХреЙрд▓рдмреИрдХ рдлрд╝рдВрдХреНрд╢рди рдФрд░ рд╡рд░реНрдЧ рдирд╛рдореЛрдВ рдХреЗ рд╕рд╛рде рдПрдХ рд╕реЗрдЧрдореЗрдВрдЯрдореЗрдЯреНрд░рд┐рдХреНрд╕ рдЙрджрд╛рд╣рд░рдг рдкреНрд░рд╛рд░рдВрдн рдХрд░реЗрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
class_result(i)
рдХрд┐рд╕реА рдирд┐рд░реНрджрд┐рд╖реНрдЯ рд╡рд░реНрдЧ рдЕрдиреБрдХреНрд░рдордгрд┐рдХрд╛ рдХреЗ рд▓рд┐рдП рд╡рд░реНрдЧреАрдХрд░рдг рдкрд░рд┐рдгрд╛рдо рджреЗрддрд╛ рд╣реИ.
mean_results()
рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдФрд░ рд╡рд┐рднрд╛рдЬрди рдкрд░рд┐рдгрд╛рдореЛрдВ рдХреЗ рд▓рд┐рдП рдорд╛рдзреНрдп рдореАрдЯреНрд░рд┐рдХ рд▓реМрдЯрд╛рдПрдВред
process(tp, tp_m, conf, pred_cls, target_cls)
рднрд╡рд┐рд╖реНрдпрд╡рд╛рдгрд┐рдпреЛрдВ рдХреЗ рджрд┐рдП рдЧрдП рд╕реЗрдЯ рдкрд░ рдкрддрд╛ рд▓рдЧрд╛рдиреЗ рдФрд░ рд╡рд┐рднрд╛рдЬрди рдореАрдЯреНрд░рд┐рдХ рдХреЛ рд╕рдВрд╕рд╛рдзрд┐рдд рдХрд░рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
tp |
list
|
рдЯреНрд░реВ рдкреЙрдЬрд┐рдЯрд┐рд╡ рдмреЙрдХреНрд╕ рдХреА рд╕реВрдЪреАред |
рдЖрд╡рд╢реНрдпрдХ |
tp_m |
list
|
рдЯреНрд░реВ рдкреЙрдЬрд┐рдЯрд┐рд╡ рдорд╛рд╕реНрдХ рдХреА рд╕реВрдЪреАред |
рдЖрд╡рд╢реНрдпрдХ |
conf |
list
|
рдЖрддреНрдорд╡рд┐рд╢реНрд╡рд╛рд╕ рд╕реНрдХреЛрд░ рдХреА рд╕реВрдЪреАред |
рдЖрд╡рд╢реНрдпрдХ |
pred_cls |
list
|
рдЕрдиреБрдорд╛рдирд┐рдд рд╡рд░реНрдЧреЛрдВ рдХреА рд╕реВрдЪреАред |
рдЖрд╡рд╢реНрдпрдХ |
target_cls |
list
|
рд▓рдХреНрд╖реНрдп рд╡рд░реНрдЧреЛрдВ рдХреА рд╕реВрдЪреАред |
рдЖрд╡рд╢реНрдпрдХ |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
ultralytics.utils.metrics.PoseMetrics
рдХрд╛ рд░реВрдк: SegmentMetrics
рдХрдХреНрд╖рд╛рдУрдВ рдХреЗ рдХрд┐рд╕реА рджрд┐рдП рдЧрдП рд╕реЗрдЯ рдкрд░ рдЧрдгрдирд╛ рдФрд░ рдкрддрд╛ рд▓рдЧрд╛рдиреЗ рдФрд░ рдореИрдЯреНрд░рд┐рдХреНрд╕ рдХреЛ рдПрдХрддреНрд░рд┐рдд рдХрд░рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
save_dir |
Path
|
рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рдХреЗ рд▓рд┐рдП рдкрде рдЬрд╣рд╛рдВ рдЖрдЙрдЯрдкреБрдЯ рдкреНрд▓реЙрдЯ рд╕рд╣реЗрдЬреЗ рдЬрд╛рдиреЗ рдЪрд╛рд╣рд┐рдПред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рд╡рд░реНрддрдорд╛рди рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рд╣реИред |
Path('.')
|
plot |
bool
|
рдЪрд╛рд╣реЗ рдкрддрд╛ рд▓рдЧрд╛рдиреЗ рдФрд░ рд╡рд┐рднрд╛рдЬрди рднреВрдЦрдВрдбреЛрдВ рдХреЛ рдмрдЪрд╛рдиреЗ рдХреЗ рд▓рд┐рдПред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рдЧрд╝рд▓рдд рд╣реИ. |
False
|
on_plot |
func
|
рдкреНрд▓реЙрдЯреНрд╕ рдкрде рдФрд░ рдбреЗрдЯрд╛ рдХреЛ рдкрд╛рд╕ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ рд╡реИрдХрд▓реНрдкрд┐рдХ рдХреЙрд▓рдмреИрдХ рдЬрдм рд╡реЗ рдкреНрд░рд╕реНрддреБрдд рдХрд┐рдП рдЬрд╛рддреЗ рд╣реИрдВред рдХреЛрдИ рдирд╣реАрдВ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
None
|
names |
list
|
рд╡рд░реНрдЧ рдХреЗ рдирд╛рдореЛрдВ рдХреА рд╕реВрдЪреАред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рдПрдХ рд░рд┐рдХреНрдд рд╕реВрдЪреА рд╣реИ. |
()
|
рд╡рд┐рд╢реЗрд╖рддрд╛рдПрдБ:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|---|
save_dir |
Path
|
рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рдХреЗ рд▓рд┐рдП рдкрде рдЬрд╣рд╛рдВ рдЖрдЙрдЯрдкреБрдЯ рдкреНрд▓реЙрдЯ рд╕рд╣реЗрдЬреЗ рдЬрд╛рдиреЗ рдЪрд╛рд╣рд┐рдПред |
plot |
bool
|
рдЪрд╛рд╣реЗ рдкрддрд╛ рд▓рдЧрд╛рдиреЗ рдФрд░ рд╡рд┐рднрд╛рдЬрди рднреВрдЦрдВрдбреЛрдВ рдХреЛ рдмрдЪрд╛рдиреЗ рдХреЗ рд▓рд┐рдПред |
on_plot |
func
|
рдкреНрд▓реЙрдЯреНрд╕ рдкрде рдФрд░ рдбреЗрдЯрд╛ рдХреЛ рдкрд╛рд╕ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ рд╡реИрдХрд▓реНрдкрд┐рдХ рдХреЙрд▓рдмреИрдХ рдЬрдм рд╡реЗ рдкреНрд░рд╕реНрддреБрдд рдХрд┐рдП рдЬрд╛рддреЗ рд╣реИрдВред |
names |
list
|
рд╡рд░реНрдЧ рдХреЗ рдирд╛рдореЛрдВ рдХреА рд╕реВрдЪреАред |
box |
Metric
|
рдмреЙрдХреНрд╕ рдбрд┐рдЯреЗрдХреНрд╢рди рдореЗрдЯреНрд░рд┐рдХреНрд╕ рдХреА рдЧрдгрдирд╛ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдореАрдЯреНрд░рд┐рдХ рд╡рд░реНрдЧ рдХрд╛ рдПрдХ рдЙрджрд╛рд╣рд░рдг. |
pose |
Metric
|
рдорд╛рд╕реНрдХ рд╡рд┐рднрд╛рдЬрди рдореАрдЯреНрд░рд┐рдХ рдХреА рдЧрдгрдирд╛ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдореАрдЯреНрд░рд┐рдХ рд╡рд░реНрдЧ рдХрд╛ рдПрдХ рдЙрджрд╛рд╣рд░рдгред |
speed |
dict
|
рдЕрдиреБрдорд╛рди рдХреЗ рд╡рд┐рднрд┐рдиреНрди рдЪрд░рдгреЛрдВ рдореЗрдВ рд▓рд┐рдП рдЧрдП рд╕рдордп рдХреЛ рд╕рдВрдЧреНрд░рд╣реАрдд рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рд╢рдмреНрджрдХреЛрд╢ред |
рд╡рд┐рдзрд┐рдпрд╛рдБ:
рдирд╛рдо | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
process |
рднрд╡рд┐рд╖реНрдпрд╡рд╛рдгрд┐рдпреЛрдВ рдХреЗ рджрд┐рдП рдЧрдП рд╕реЗрдЯ рдкрд░ рдореИрдЯреНрд░рд┐рдХреНрд╕ рд╕рдВрд╕рд╛рдзрд┐рдд рдХрд░рддрд╛ рд╣реИред |
mean_results |
рд╕рднреА рд╡рд░реНрдЧреЛрдВ рдкрд░ рдкрддрд╛ рд▓рдЧрд╛рдиреЗ рдФрд░ рд╡рд┐рднрд╛рдЬрди рдореИрдЯреНрд░рд┐рдХреНрд╕ рдХрд╛ рдорд╛рдзреНрдп рджреЗрддрд╛ рд╣реИред |
class_result |
рд╡рд░реНрдЧ рдХреА рдЦреЛрдЬ рдФрд░ рд╡рд┐рднрд╛рдЬрди рдореИрдЯреНрд░рд┐рдХреНрд╕ рд▓реМрдЯрд╛рддрд╛ рд╣реИ |
maps |
0.50 рд╕реЗ 0.95 рддрдХ рдХреА IoU рдереНрд░реЗрд╢реЛрд▓реНрдб рдХреЗ рд▓рд┐рдП рдорд╛рдзреНрдп рдФрд╕рдд рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ (mAP) рд╕реНрдХреЛрд░ рджреЗрддрд╛ рд╣реИ. |
fitness |
рдлрд╝рд┐рдЯрдиреЗрд╕ рд╕реНрдХреЛрд░ рд▓реМрдЯрд╛рддрд╛ рд╣реИ, рдЬреЛ рдореЗрдЯреНрд░рд┐рдХреНрд╕ рдХрд╛ рдПрдХрд▓ рднрд╛рд░рд┐рдд рд╕рдВрдпреЛрдЬрди рд╣реЛрддрд╛ рд╣реИ. |
ap_class_index |
рдФрд╕рдд рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ (AP) рдХреА рдЧрдгрдирд╛ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдЙрдкрдпреЛрдЧ рдХрд┐рдП рдЬрд╛рдиреЗ рд╡рд╛рд▓реЗ рд╡рд░реНрдЧреЛрдВ рдХреЗ рд╕реВрдЪрдХрд╛рдВрдХреЛрдВ рдХреА рд╕реВрдЪреА рджреЗрддрд╛ рд╣реИ. |
results_dict |
рд╕рднреА рдкрд╣рдЪрд╛рди рдФрд░ рд╡рд┐рднрд╛рдЬрди рдореИрдЯреНрд░рд┐рдХреНрд╕ рдФрд░ рдлрд┐рдЯрдиреЗрд╕ рд╕реНрдХреЛрд░ рдпреБрдХреНрдд рд╢рдмреНрджрдХреЛрд╢ рджреЗрддрд╛ рд╣реИ. |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 |
|
curves
property
рд╡рд┐рд╢рд┐рд╖реНрдЯ рдореАрдЯреНрд░рд┐рдХ рд╡рдХреНрд░реЛрдВ рддрдХ рдкрд╣реБрдБрдЪрдиреЗ рдХреЗ рд▓рд┐рдП рд╡рдХреНрд░реЛрдВ рдХреА рдПрдХ рд╕реВрдЪреА рджреЗрддрд╛ рд╣реИред
curves_results
property
рдЧрдгрдирд╛ рдХрд┐рдП рдЧрдП рдкреНрд░рджрд░реНрд╢рди рдореИрдЯреНрд░рд┐рдХреНрд╕ рдФрд░ рдЖрдВрдХрдбрд╝реЛрдВ рдХрд╛ рд╢рдмреНрджрдХреЛрд╢ рджреЗрддрд╛ рд╣реИред
fitness
property
рд╡рд░реНрдЧреАрдХрд░рдг рдореАрдЯреНрд░рд┐рдХ рдФрд░ рдЧрддрд┐ рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░рдХреЗ рдЧрдгрдирд╛ рдХрд░рддрд╛ рд╣реИ targets
рдФрд░ pred
рдЬрд╛рдирдХрд╛рд░реАред
keys
property
рдореВрд▓реНрдпрд╛рдВрдХрди рдореАрдЯреНрд░рд┐рдХ рдХреБрдВрдЬреА рдХреА рд╕реВрдЪреА рд▓реМрдЯрд╛рддрд╛ рд╣реИред
maps
property
рдмреЙрдХреНрд╕ рдФрд░ рдореБрджреНрд░рд╛ рдбрд┐рдЯреЗрдХреНрд╢рди рджреЛрдиреЛрдВ рдХреЗ рд▓рд┐рдП рдкреНрд░рддрд┐ рд╡рд░реНрдЧ рдорд╛рдзреНрдп рдФрд╕рдд рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ (mAP) рджреЗрддрд╛ рд╣реИ.
__init__(save_dir=Path('.'), plot=False, on_plot=None, names=())
PoseMetrics рдХреНрд▓рд╛рд╕ рдХреЛ рдбрд╛рдпрд░реЗрдХреНрдЯрд░реА рдкрде, рдХреНрд▓рд╛рд╕ рдирд╛рдо рдФрд░ рдкреНрд▓реЙрдЯрд┐рдВрдЧ рд╡рд┐рдХрд▓реНрдкреЛрдВ рдХреЗ рд╕рд╛рде рдЗрдирд┐рд╢рд┐рдпрд▓рд╛рдЗрдЬрд╝ рдХрд░реЗрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
class_result(i)
рдПрдХ рд╡рд┐рд╢рд┐рд╖реНрдЯ рд╡рд░реНрдЧ i рдХреЗ рд▓рд┐рдП рдХрдХреНрд╖рд╛рд╡рд╛рд░ рдкрд╣рдЪрд╛рди рдкрд░рд┐рдгрд╛рдо рд▓реМрдЯрд╛рдПрдВред
mean_results()
рдмреЙрдХреНрд╕ рдФрд░ рдореБрджреНрд░рд╛ рдХреЗ рдорд╛рдзреНрдп рдкрд░рд┐рдгрд╛рдо рд▓реМрдЯрд╛рдПрдВред
process(tp, tp_p, conf, pred_cls, target_cls)
рднрд╡рд┐рд╖реНрдпрд╡рд╛рдгрд┐рдпреЛрдВ рдХреЗ рджрд┐рдП рдЧрдП рд╕реЗрдЯ рдкрд░ рдкрддрд╛ рд▓рдЧрд╛рдиреЗ рдФрд░ рдореЗрдЯреНрд░рд┐рдХреНрд╕ рдХреЛ рд╕рдВрд╕рд╛рдзрд┐рдд рдХрд░рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
tp |
list
|
рдЯреНрд░реВ рдкреЙрдЬрд┐рдЯрд┐рд╡ рдмреЙрдХреНрд╕ рдХреА рд╕реВрдЪреАред |
рдЖрд╡рд╢реНрдпрдХ |
tp_p |
list
|
рд╕рдЪреНрдЪреЗ рд╕рдХрд╛рд░рд╛рддреНрдордХ рдХреАрдкреЙрдЗрдВрдЯреНрд╕ рдХреА рд╕реВрдЪреАред |
рдЖрд╡рд╢реНрдпрдХ |
conf |
list
|
рдЖрддреНрдорд╡рд┐рд╢реНрд╡рд╛рд╕ рд╕реНрдХреЛрд░ рдХреА рд╕реВрдЪреАред |
рдЖрд╡рд╢реНрдпрдХ |
pred_cls |
list
|
рдЕрдиреБрдорд╛рдирд┐рдд рд╡рд░реНрдЧреЛрдВ рдХреА рд╕реВрдЪреАред |
рдЖрд╡рд╢реНрдпрдХ |
target_cls |
list
|
рд▓рдХреНрд╖реНрдп рд╡рд░реНрдЧреЛрдВ рдХреА рд╕реВрдЪреАред |
рдЖрд╡рд╢реНрдпрдХ |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
ultralytics.utils.metrics.ClassifyMetrics
рдХрд╛ рд░реВрдк: SimpleClass
рд╢реАрд░реНрд╖ -1 рдФрд░ рд╢реАрд░реНрд╖ -5 рд╕рдЯреАрдХрддрд╛ рд╕рд╣рд┐рдд рдХрдВрдкреНрдпреВрдЯрд┐рдВрдЧ рд╡рд░реНрдЧреАрдХрд░рдг рдореИрдЯреНрд░рд┐рдХреНрд╕ рдХреЗ рд▓рд┐рдП рд╡рд░реНрдЧред
рд╡рд┐рд╢реЗрд╖рддрд╛рдПрдБ:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|---|
top1 |
float
|
рд╢реАрд░реНрд╖ -1 рд╕рдЯреАрдХрддрд╛ред |
top5 |
float
|
рд╢реАрд░реНрд╖ -5 рд╕рдЯреАрдХрддрд╛ред |
speed |
Dict[str, float]
|
рдПрдХ рд╢рдмреНрджрдХреЛрд╢ рдЬрд┐рд╕рдореЗрдВ рдкрд╛рдЗрдкрд▓рд╛рдЗрди рдореЗрдВ рдкреНрд░рддреНрдпреЗрдХ рдЪрд░рдг рдХреЗ рд▓рд┐рдП рд▓рд┐рдпрд╛ рдЧрдпрд╛ рд╕рдордп рдирд┐рд╣рд┐рдд рд╣реИред |
рдЧреБрдг
рдлрд┐рдЯрдиреЗрд╕ (рдлреНрд▓реЛрдЯ): рдореЙрдбрд▓ рдХреА рдлрд┐рдЯрдиреЗрд╕, рдЬреЛ рдЯреЙрдк-5 рдПрдХреНрдпреВрд░реЗрд╕реА рдХреЗ рдмрд░рд╛рдмрд░ рд╣реИред results_dict (рдбрд┐рдХреНрдЯ [рдПрд╕рдЯреАрдЖрд░, рдпреВрдирд┐рдпрди [рдлреНрд▓реЛрдЯ, рдПрд╕рдЯреАрдЖрд░]]): рд╡рд░реНрдЧреАрдХрд░рдг рдореИрдЯреНрд░рд┐рдХреНрд╕ рдФрд░ рдлрд┐рдЯрдиреЗрд╕ рдпреБрдХреНрдд рдПрдХ рд╢рдмреНрджрдХреЛрд╢ред keys (List [str]): results_dict рдХреЗ рд▓рд┐рдП рдХреБрдВрдЬрд┐рдпреЛрдВ рдХреА рдПрдХ рд╕реВрдЪреАред
рд╡рд┐рдзрд┐рдпрд╛рдБ:
рдирд╛рдо | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
process |
рд╡рд░реНрдЧреАрдХрд░рдг рдореАрдЯреНрд░рд┐рдХ рдХреА рдЧрдгрдирд╛ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рд▓рдХреНрд╖реНрдпреЛрдВ рдФрд░ рднрд╡рд┐рд╖реНрдпрд╡рд╛рдгрд┐рдпреЛрдВ рдХреЛ рд╕рдВрд╕рд╛рдзрд┐рдд рдХрд░рддрд╛ рд╣реИред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
curves
property
рд╡рд┐рд╢рд┐рд╖реНрдЯ рдореАрдЯреНрд░рд┐рдХ рд╡рдХреНрд░реЛрдВ рддрдХ рдкрд╣реБрдБрдЪрдиреЗ рдХреЗ рд▓рд┐рдП рд╡рдХреНрд░реЛрдВ рдХреА рдПрдХ рд╕реВрдЪреА рджреЗрддрд╛ рд╣реИред
curves_results
property
рд╡рд┐рд╢рд┐рд╖реНрдЯ рдореАрдЯреНрд░рд┐рдХ рд╡рдХреНрд░реЛрдВ рддрдХ рдкрд╣реБрдБрдЪрдиреЗ рдХреЗ рд▓рд┐рдП рд╡рдХреНрд░реЛрдВ рдХреА рдПрдХ рд╕реВрдЪреА рджреЗрддрд╛ рд╣реИред
fitness
property
рд░рд┐рдЯрд░реНрди рдХрд╛ рдорддрд▓рдм рдлрд┐рдЯрдиреЗрд╕ рд╕реНрдХреЛрд░ рдХреЗ рд░реВрдк рдореЗрдВ рд╢реАрд░реНрд╖ -1 рдФрд░ рд╢реАрд░реНрд╖ -5 рд╕рдЯреАрдХрддрд╛ рд╣реИред
keys
property
results_dict рдЧреБрдг рдХреЗ рд▓рд┐рдП рдХреБрдВрдЬрд┐рдпреЛрдВ рдХреА рд╕реВрдЪреА рд▓реМрдЯрд╛рддрд╛ рд╣реИ.
results_dict
property
рдореЙрдбрд▓ рдХреЗ рдкреНрд░рджрд░реНрд╢рди рдореАрдЯреНрд░рд┐рдХ рдФрд░ рдлрд╝рд┐рдЯрдиреЗрд╕ рд╕реНрдХреЛрд░ рдХреЗ рд╕рд╛рде рдПрдХ рд╢рдмреНрджрдХреЛрд╢ рд▓реМрдЯрд╛рддрд╛ рд╣реИ.
__init__()
ClassifyMetrics рдЗрдВрд╕реНрдЯреЗрдВрд╕ рд╢реБрд░реВ рдХрд░реЗрдВ.
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
process(targets, pred)
рд▓рдХреНрд╖реНрдп рдХрдХреНрд╖рд╛рдПрдВ рдФрд░ рдЕрдиреБрдорд╛рдирд┐рдд рдХрдХреНрд╖рд╛рдПрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
ultralytics.utils.metrics.OBBMetrics
рдХрд╛ рд░реВрдк: SimpleClass
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 |
|
ap_class_index
property
рдкреНрд░рддрд┐ рд╡рд░реНрдЧ рдФрд╕рдд рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ рд╕реВрдЪрдХрд╛рдВрдХ рджреЗрддрд╛ рд╣реИ.
curves
property
рд╡рд┐рд╢рд┐рд╖реНрдЯ рдореАрдЯреНрд░рд┐рдХ рд╡рдХреНрд░реЛрдВ рддрдХ рдкрд╣реБрдБрдЪрдиреЗ рдХреЗ рд▓рд┐рдП рд╡рдХреНрд░реЛрдВ рдХреА рдПрдХ рд╕реВрдЪреА рджреЗрддрд╛ рд╣реИред
curves_results
property
рд╡рд┐рд╢рд┐рд╖реНрдЯ рдореАрдЯреНрд░рд┐рдХ рд╡рдХреНрд░реЛрдВ рддрдХ рдкрд╣реБрдБрдЪрдиреЗ рдХреЗ рд▓рд┐рдП рд╡рдХреНрд░реЛрдВ рдХреА рдПрдХ рд╕реВрдЪреА рджреЗрддрд╛ рд╣реИред
fitness
property
рдмреЙрдХреНрд╕ рдСрдмреНрдЬреЗрдХреНрдЯ рдХреА рдлрд┐рдЯрдиреЗрд╕ рд▓реМрдЯрд╛рддрд╛ рд╣реИред
keys
property
рд╡рд┐рд╢рд┐рд╖реНрдЯ рдореАрдЯреНрд░рд┐рдХ рддрдХ рдкрд╣реБрдБрдЪрдиреЗ рдХреЗ рд▓рд┐рдП рдХреБрдВрдЬрд┐рдпреЛрдВ рдХреА рдПрдХ рд╕реВрдЪреА рджреЗрддрд╛ рд╣реИред
maps
property
рд░рд┐рдЯрд░реНрди рдХрд╛ рдорддрд▓рдм рдкреНрд░рддрд┐ рд╡рд░реНрдЧ рдФрд╕рдд рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ (рдПрдордПрдкреА) рд╕реНрдХреЛрд░ рд╣реИред
results_dict
property
рдЧрдгрдирд╛ рдХрд┐рдП рдЧрдП рдкреНрд░рджрд░реНрд╢рди рдореИрдЯреНрд░рд┐рдХреНрд╕ рдФрд░ рдЖрдВрдХрдбрд╝реЛрдВ рдХрд╛ рд╢рдмреНрджрдХреЛрд╢ рджреЗрддрд╛ рд╣реИред
class_result(i)
рдПрдХ рд╡рд┐рд╢рд┐рд╖реНрдЯ рд╡рд░реНрдЧ рдкрд░ рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди рдореЙрдбрд▓ рдХреЗ рдкреНрд░рджрд░реНрд╢рди рдХрд╛ рдореВрд▓реНрдпрд╛рдВрдХрди рдХрд░рдиреЗ рдХрд╛ рдкрд░рд┐рдгрд╛рдо рд▓реМрдЯрд╛рдПрдВред
mean_results()
рдкрддрд╛ рдЪрд▓рд╛ рд╡рд╕реНрддреБрдУрдВ рдХрд╛ рдорддрд▓рдм рдХреА рдЧрдгрдирд╛ & рд╡рд╛рдкрд╕реА рдкрд░рд┐рд╢реБрджреНрдзрддрд╛, рдпрд╛рдж, mAP50, рдФрд░ mAP50-95.
process(tp, conf, pred_cls, target_cls)
рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди рдФрд░ рдЕрдкрдбреЗрдЯ рдореЗрдЯреНрд░рд┐рдХреНрд╕ рдХреЗ рд▓рд┐рдП рдЕрдиреБрдорд╛рдирд┐рдд рдкрд░рд┐рдгрд╛рдореЛрдВ рдХреА рдкреНрд░рдХреНрд░рд┐рдпрд╛ рдХрд░реЗрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
ultralytics.utils.metrics.bbox_ioa(box1, box2, iou=False, eps=1e-07)
рджрд┐рдП рдЧрдП рдмреЙрдХреНрд╕ 2 рдХреНрд╖реЗрддреНрд░ рдкрд░ рдЪреМрд░рд╛рд╣реЗ рдХреА рдЧрдгрдирд╛ рдХрд░реЗрдВ рдмреЙрдХреНрд╕ 1 рдФрд░ рдмреЙрдХреНрд╕ 2ред рдмреЙрдХреНрд╕ x1y1x2y2 рдкреНрд░рд╛рд░реВрдк рдореЗрдВ рд╣реИрдВред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
box1 |
ndarray
|
рдЖрдХрд╛рд░ рдХреА рдПрдХ рд╕реБрдиреНрди рд╕рд░рдгреА (рдПрди, 4) рдЬреЛ рдПрди рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдХрд╛ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ рдХрд░рддреА рд╣реИред |
рдЖрд╡рд╢реНрдпрдХ |
box2 |
ndarray
|
рдЖрдХрд╛рд░ рдХреА рдПрдХ рд╕реБрдиреНрди рд╕рд░рдгреА (рдПрдо, 4) рдЬреЛ рдПрдо рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдХрд╛ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ рдХрд░рддреА рд╣реИред |
рдЖрд╡рд╢реНрдпрдХ |
iou |
bool
|
рдорд╛рдирдХ IoU рдХреА рдЧрдгрдирд╛ рдХрд░реЗрдВ рдпрджрд┐ рд╕рд╣реА рд╣реИ рдЕрдиреНрдпрдерд╛ inter_area/box2_area рд▓реМрдЯрд╛рдПрдВред |
False
|
eps |
float
|
рд╢реВрдиреНрдп рд╕реЗ рд╡рд┐рднрд╛рдЬрди рд╕реЗ рдмрдЪрдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ рдЫреЛрдЯрд╛ рдореВрд▓реНрдпред 1e-7 рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
1e-07
|
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
ndarray
|
рдЖрдХрд╛рд░ рдХреА рдПрдХ рд╕реБрдиреНрди рд╕рд░рдгреА (n, m) рдмреЙрдХреНрд╕ 2 рдХреНрд╖реЗрддреНрд░ рдкрд░ рдЪреМрд░рд╛рд╣реЗ рдХрд╛ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ рдХрд░рддреА рд╣реИред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
ultralytics.utils.metrics.box_iou(box1, box2, eps=1e-07)
рдмрдХреНрд╕реЗ рдХреЗ рдЪреМрд░рд╛рд╣реЗ-рдУрд╡рд░-рдпреВрдирд┐рдпрди (IoU) рдХреА рдЧрдгрдирд╛ рдХрд░реЗрдВред рдмреЙрдХреНрд╕ рдХреЗ рджреЛрдиреЛрдВ рд╕реЗрдЯ (x1, y1, x2, y2) рдкреНрд░рд╛рд░реВрдк рдореЗрдВ рд╣реЛрдиреЗ рдХреА рдЙрдореНрдореАрдж рд╣реИред https://github.com/ рдЖрдзрд╛рд░ рдкрд░pytorch/рджреГрд╖реНрдЯрд┐/рдмреВрдБрдж/рдорд╛рд╕реНрдЯрд░/рдорд╢рд╛рд▓рд╡рд┐рдЬрди/рдСрдкреНрд╕/boxes.py
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
box1 |
Tensor
|
рдПрдХ tensor рдЖрдХрд╛рд░ рдХрд╛ (рдПрди, 4) рдПрди рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдХрд╛ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ рдХрд░рддрд╛ рд╣реИред |
рдЖрд╡рд╢реНрдпрдХ |
box2 |
Tensor
|
рдПрдХ tensor рдЖрдХрд╛рд░ рдХрд╛ (рдПрдо, 4) рдПрдо рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдХрд╛ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ рдХрд░рддрд╛ рд╣реИред |
рдЖрд╡рд╢реНрдпрдХ |
eps |
float
|
рд╢реВрдиреНрдп рд╕реЗ рд╡рд┐рднрд╛рдЬрди рд╕реЗ рдмрдЪрдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ рдЫреЛрдЯрд╛ рдореВрд▓реНрдпред 1e-7 рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
1e-07
|
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
Tensor
|
рдПрдХ рдПрдирдПрдХреНрд╕рдПрдо tensor рдЬрд┐рд╕рдореЗрдВ рдмреЙрдХреНрд╕ 1 рдФрд░ рдмреЙрдХреНрд╕ 2 рдореЗрдВ рдкреНрд░рддреНрдпреЗрдХ рддрддреНрд╡ рдХреЗ рд▓рд┐рдП рдЬреЛрдбрд╝реАрджрд╛рд░ IoU рдорд╛рди рд╢рд╛рдорд┐рд▓ рд╣реИрдВред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
ultralytics.utils.metrics.bbox_iou(box1, box2, xywh=True, GIoU=False, DIoU=False, CIoU=False, eps=1e-07)
рдмреЙрдХреНрд╕ 1 (1, 4) рд╕реЗ рдмреЙрдХреНрд╕ 2 (рдПрди, 4) рдХреЗ рд╕рдВрдШ (рдЖрдИрдУрдпреВ) рдкрд░ рдЪреМрд░рд╛рд╣реЗ рдХреА рдЧрдгрдирд╛ рдХрд░реЗрдВред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
box1 |
Tensor
|
рдПрдХ tensor рдЖрдХрд╛рд░ (1, 4) рдХреЗ рд╕рд╛рде рдПрдХ рдПрдХрд▓ рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдХрд╛ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ рдХрд░рдирд╛ред |
рдЖрд╡рд╢реНрдпрдХ |
box2 |
Tensor
|
рдПрдХ tensor рдЖрдХрд╛рд░ рдХреЗ рд╕рд╛рде рдПрди рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмрдХреНрд╕реЗ рдХрд╛ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ (рдПрди, 4)ред |
рдЖрд╡рд╢реНрдпрдХ |
xywh |
bool
|
рдпрджрд┐ True рд╣реИ, рддреЛ рдЗрдирдкреБрдЯ рдмреЙрдХреНрд╕ (x, y, w, h) рд╕реНрд╡рд░реВрдк рдореЗрдВ рд╣реИрдВред рдпрджрд┐ рдЧрд▓рдд рд╣реИ, рддреЛ рдЗрдирдкреБрдЯ рдмреЙрдХреНрд╕ (x1, y1, x2, y2) рдкреНрд░рд╛рд░реВрдкред рд╕рд╣реА рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
True
|
GIoU |
bool
|
рдпрджрд┐ рд╕рд╣реА рд╣реИ, рддреЛ рд╕рд╛рдорд╛рдиреНрдпреАрдХреГрдд IoU рдХреА рдЧрдгрдирд╛ рдХрд░реЗрдВред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рд░реВрдк рд╕реЗ рдЧрд▓рдд рд╣реИ. |
False
|
DIoU |
bool
|
рдпрджрд┐ рд╕рд╣реА рд╣реИ, рддреЛ рджреВрд░реА IoU рдХреА рдЧрдгрдирд╛ рдХрд░реЗрдВред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рд░реВрдк рд╕реЗ рдЧрд▓рдд рд╣реИ. |
False
|
CIoU |
bool
|
рдпрджрд┐ рд╕рд╣реА рд╣реИ, рддреЛ рдкреВрд░реНрдг IoU рдХреА рдЧрдгрдирд╛ рдХрд░реЗрдВред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рд░реВрдк рд╕реЗ рдЧрд▓рдд рд╣реИ. |
False
|
eps |
float
|
рд╢реВрдиреНрдп рд╕реЗ рд╡рд┐рднрд╛рдЬрди рд╕реЗ рдмрдЪрдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ рдЫреЛрдЯрд╛ рдореВрд▓реНрдпред 1e-7 рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
1e-07
|
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
Tensor
|
рдирд┐рд░реНрджрд┐рд╖реНрдЯ рдЭрдВрдбреЗ рдХреЗ рдЖрдзрд╛рд░ рдкрд░ IoU, GIoU, DIoU, рдпрд╛ CIoU рдорд╛рдиред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
ultralytics.utils.metrics.mask_iou(mask1, mask2, eps=1e-07)
рдорд╛рд╕реНрдХ IoU рдХреА рдЧрдгрдирд╛ рдХрд░реЗрдВред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
mask1 |
Tensor
|
рдПрдХ tensor рдЖрдХрд╛рд░ рдХрд╛ (N, n) рдЬрд╣рд╛рдБ N рдЬрдореАрдиреА рд╕рддреНрдп рд╡рд╕реНрддреБрдУрдВ рдХреА рд╕рдВрдЦреНрдпрд╛ рд╣реИ рдФрд░ n рд╣реИ рдЫрд╡рд┐ рдЪреМрдбрд╝рд╛рдИ рдФрд░ рдКрдВрдЪрд╛рдИ рдХрд╛ рдЙрддреНрдкрд╛рджред |
рдЖрд╡рд╢реНрдпрдХ |
mask2 |
Tensor
|
рдПрдХ tensor рдЖрдХрд╛рд░ рдХрд╛ (M, n) рдЬрд╣рд╛рдБ M рдкреВрд░реНрд╡рд╛рдиреБрдорд╛рдирд┐рдд рд╡рд╕реНрддреБрдУрдВ рдХреА рд╕рдВрдЦреНрдпрд╛ рд╣реИ рдФрд░ n рд╣реИ рдЫрд╡рд┐ рдЪреМрдбрд╝рд╛рдИ рдФрд░ рдКрдВрдЪрд╛рдИ рдХрд╛ рдЙрддреНрдкрд╛рджред |
рдЖрд╡рд╢реНрдпрдХ |
eps |
float
|
рд╢реВрдиреНрдп рд╕реЗ рд╡рд┐рднрд╛рдЬрди рд╕реЗ рдмрдЪрдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ рдЫреЛрдЯрд╛ рдореВрд▓реНрдпред 1e-7 рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
1e-07
|
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
Tensor
|
рдПрдХ tensor рдЖрдХрд╛рд░ рдХрд╛ (рдПрди, рдПрдо) рдорд╛рд╕реНрдХ IoU рдХрд╛ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ рдХрд░рддрд╛ рд╣реИред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
ultralytics.utils.metrics.kpt_iou(kpt1, kpt2, area, sigma, eps=1e-07)
рдСрдмреНрдЬреЗрдХреНрдЯ рдХреАрдкреЙрдЗрдВрдЯ рд╕рдорд╛рдирддрд╛ (рдУрдХреЗрдПрд╕) рдХреА рдЧрдгрдирд╛ рдХрд░реЗрдВред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
kpt1 |
Tensor
|
рдПрдХ tensor рдЖрдХрд╛рд░ рдХрд╛ (рдПрди, 17, 3) рдЬрдореАрдиреА рд╕рдЪреНрдЪрд╛рдИ рдХреАрдкреЙрдЗрдВрдЯреНрд╕ рдХрд╛ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ рдХрд░рддрд╛ рд╣реИред |
рдЖрд╡рд╢реНрдпрдХ |
kpt2 |
Tensor
|
рдПрдХ tensor рдЖрдХрд╛рд░ рдХрд╛ (рдПрдо, 17, 3) рдЕрдиреБрдорд╛рдирд┐рдд рдХреАрдкреЙрдЗрдВрдЯреНрд╕ рдХрд╛ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ рдХрд░рддрд╛ рд╣реИред |
рдЖрд╡рд╢реНрдпрдХ |
area |
Tensor
|
рдПрдХ tensor рдЖрдХрд╛рд░ рдХрд╛ (рдПрди,) рдЬрдореАрдиреА рд╕рдЪреНрдЪрд╛рдИ рд╕реЗ рдХреНрд╖реЗрддреНрд░реЛрдВ рдХрд╛ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ рдХрд░рддрд╛ рд╣реИред |
рдЖрд╡рд╢реНрдпрдХ |
sigma |
list
|
рдПрдХ рд╕реВрдЪреА рдЬрд┐рд╕рдореЗрдВ рдореБрдЦреНрдп рдмрд┐рдВрджреБ рдкреИрдорд╛рдиреЛрдВ рдХрд╛ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ рдХрд░рдиреЗ рд╡рд╛рд▓реЗ 17 рдорд╛рди рд╣реЛрддреЗ рд╣реИрдВ. |
рдЖрд╡рд╢реНрдпрдХ |
eps |
float
|
рд╢реВрдиреНрдп рд╕реЗ рд╡рд┐рднрд╛рдЬрди рд╕реЗ рдмрдЪрдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ рдЫреЛрдЯрд╛ рдореВрд▓реНрдпред 1e-7 рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
1e-07
|
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
Tensor
|
рдПрдХ tensor рдЖрдХрд╛рд░ рдХрд╛ (рдПрди, рдПрдо) рдореБрдЦреНрдп рдмрд┐рдВрджреБ рд╕рдорд╛рдирддрд╛ рдХрд╛ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ рдХрд░рддрд╛ рд╣реИред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
ultralytics.utils.metrics._get_covariance_matrix(boxes)
obbs рд╕реЗ рд╕рд╣рдкреНрд░рд╕рд░рдг рдореИрдЯреНрд░рд┐рдХреНрд╕ рдЙрддреНрдкрдиреНрди рдХрд░рдирд╛ред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
boxes |
Tensor
|
рдПрдХ tensor рдЖрдХрд╛рд░ рдХрд╛ (рдПрди, 5) рдШреБрдорд╛рдП рдЧрдП рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдХрд╛ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ рдХрд░рддрд╛ рд╣реИ, xywhr рдкреНрд░рд╛рд░реВрдк рдХреЗ рд╕рд╛рдеред |
рдЖрд╡рд╢реНрдпрдХ |
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
Tensor
|
рдореВрд▓ рдШреБрдорд╛рдП рдЧрдП рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдХреЗ рдЕрдиреБрд░реВрдк рд╕рд╣рдкреНрд░рд╕рд░рдг рдореЗрдЯреНрд░рд┐рдХреНрд╕ред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
ultralytics.utils.metrics.probiou(obb1, obb2, CIoU=False, eps=1e-07)
рдЙрдиреНрдореБрдЦ рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмрдХреНрд╕реЗ, https://arxiv.org/pdf/2106.06072v1.pdf рдХреЗ рдмреАрдЪ рд╕рдорд╕реНрдпрд╛ IoU рдХреА рдЧрдгрдирд╛ рдХрд░реЗрдВред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
obb1 |
Tensor
|
рдПрдХ tensor рдЖрдХрд╛рд░ рдХрд╛ (рдПрди, 5) рдЬрдореАрдиреА рд╕рдЪреНрдЪрд╛рдИ рдУрдмреНрд╕ рдХрд╛ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ рдХрд░рддрд╛ рд╣реИ, xywhr рдкреНрд░рд╛рд░реВрдк рдХреЗ рд╕рд╛рдеред |
рдЖрд╡рд╢реНрдпрдХ |
obb2 |
Tensor
|
рдПрдХ tensor рдЖрдХрд╛рд░ рдХрд╛ (рдПрди, 5) рдЕрдиреБрдорд╛рдирд┐рдд рдУрдмреНрд╕ рдХрд╛ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ рдХрд░рддрд╛ рд╣реИ, xywhr рдкреНрд░рд╛рд░реВрдк рдХреЗ рд╕рд╛рдеред |
рдЖрд╡рд╢реНрдпрдХ |
eps |
float
|
рд╢реВрдиреНрдп рд╕реЗ рд╡рд┐рднрд╛рдЬрди рд╕реЗ рдмрдЪрдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ рдЫреЛрдЯрд╛ рдореВрд▓реНрдпред 1e-7 рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
1e-07
|
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
Tensor
|
рдПрдХ tensor рдЖрдХрд╛рд░ рдХрд╛ (рдПрди, ) рдУрдмреАрдмреА рд╕рдорд╛рдирддрд╛ рдХрд╛ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ рдХрд░рддрд╛ рд╣реИред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
ultralytics.utils.metrics.batch_probiou(obb1, obb2, eps=1e-07)
рдЙрдиреНрдореБрдЦ рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмрдХреНрд╕реЗ, https://arxiv.org/pdf/2106.06072v1.pdf рдХреЗ рдмреАрдЪ рд╕рдорд╕реНрдпрд╛ IoU рдХреА рдЧрдгрдирд╛ рдХрд░реЗрдВред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
obb1 |
Tensor | ndarray
|
рдПрдХ tensor рдЖрдХрд╛рд░ рдХрд╛ (рдПрди, 5) рдЬрдореАрдиреА рд╕рдЪреНрдЪрд╛рдИ рдУрдмреНрд╕ рдХрд╛ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ рдХрд░рддрд╛ рд╣реИ, xywhr рдкреНрд░рд╛рд░реВрдк рдХреЗ рд╕рд╛рдеред |
рдЖрд╡рд╢реНрдпрдХ |
obb2 |
Tensor | ndarray
|
рдПрдХ tensor рдЖрдХрд╛рд░ рдХрд╛ (рдПрдо, 5) рдЕрдиреБрдорд╛рдирд┐рдд рдУрдмреНрд╕ рдХрд╛ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ рдХрд░рддрд╛ рд╣реИ, xywhr рдкреНрд░рд╛рд░реВрдк рдХреЗ рд╕рд╛рдеред |
рдЖрд╡рд╢реНрдпрдХ |
eps |
float
|
рд╢реВрдиреНрдп рд╕реЗ рд╡рд┐рднрд╛рдЬрди рд╕реЗ рдмрдЪрдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ рдЫреЛрдЯрд╛ рдореВрд▓реНрдпред 1e-7 рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
1e-07
|
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
Tensor
|
рдПрдХ tensor рдЖрдХрд╛рд░ рдХрд╛ (рдПрди, рдПрдо) рдУрдмреАрдмреА рд╕рдорд╛рдирддрд╛ рдХрд╛ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ рдХрд░рддрд╛ рд╣реИред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
ultralytics.utils.metrics.smooth_BCE(eps=0.1)
рдЧрдгрдирд╛ рдиреЗ рд╕рдХрд╛рд░рд╛рддреНрдордХ рдФрд░ рдирдХрд╛рд░рд╛рддреНрдордХ рдмрд╛рдЗрдирд░реА рдХреНрд░реЙрд╕-рдПрдиреНрдЯреНрд░реЙрдкреА рд▓рдХреНрд╖реНрдпреЛрдВ рдХреЛ рд╕реБрдЪрд╛рд░реВ рдХрд┐рдпрд╛ред
рдпрд╣ рдлрд╝рдВрдХреНрд╢рди рдХрд┐рд╕реА рджрд┐рдП рдЧрдП рдПрдкреНрд╕рд┐рд▓реЙрди рдореВрд▓реНрдп рдХреЗ рдЖрдзрд╛рд░ рдкрд░ рд╕рдХрд╛рд░рд╛рддреНрдордХ рдФрд░ рдирдХрд╛рд░рд╛рддреНрдордХ рд▓реЗрдмрд▓ рдЪреМрд░рд╕рд╛рдИ рдмреАрд╕реАрдИ рд▓рдХреНрд╖реНрдпреЛрдВ рдХреА рдЧрдгрдирд╛ рдХрд░рддрд╛ рд╣реИред рдХрд╛рд░реНрдпрд╛рдиреНрд╡рдпрди рд╡рд┐рд╡рд░рдг рдХреЗ рд▓рд┐рдП, https://github.com/ рджреЗрдЦреЗрдВultralytics/yolov3/issues/238#issuecomment -598028441.
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
eps |
float
|
рд▓реЗрдмрд▓ рдЪреМрд░рд╕рд╛рдИ рдХреЗ рд▓рд┐рдП рдПрдкреНрд╕рд┐рд▓реЙрди рдорд╛рдиред 0.1 рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
0.1
|
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
tuple
|
рдПрдХ рдЯрдкрд▓ рдЬрд┐рд╕рдореЗрдВ рд╕рдХрд╛рд░рд╛рддреНрдордХ рдФрд░ рдирдХрд╛рд░рд╛рддреНрдордХ рд▓реЗрдмрд▓ рдмреАрд╕реАрдИ рд▓рдХреНрд╖реНрдпреЛрдВ рдХреЛ рдЪреМрд░рд╕рд╛рдИ рдХрд░рддреЗ рд╣реИрдВред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
ultralytics.utils.metrics.smooth(y, f=0.05)
рдЕрдВрд╢ f рдХрд╛ рдмреЙрдХреНрд╕ рдлрд╝рд┐рд▓реНрдЯрд░ред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
ultralytics.utils.metrics.plot_pr_curve(px, py, ap, save_dir=Path('pr_curve.png'), names=(), on_plot=None)
рдПрдХ рд╕рдЯреАрдХ-рдпрд╛рдж рд╡рдХреНрд░ рдкреНрд▓реЙрдЯред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
ultralytics.utils.metrics.plot_mc_curve(px, py, save_dir=Path('mc_curve.png'), names=(), xlabel='Confidence', ylabel='Metric', on_plot=None)
рдПрдХ рдореАрдЯреНрд░рд┐рдХ-рдЖрддреНрдорд╡рд┐рд╢реНрд╡рд╛рд╕ рд╡рдХреНрд░ рдкреНрд▓реЙрдЯред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
ultralytics.utils.metrics.compute_ap(recall, precision)
рдпрд╛рдж рдФрд░ рд╕рдЯреАрдХ рдШрдЯрддрд╛ рджрд┐рдпрд╛ рдФрд╕рдд рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ (AP) рдХреА рдЧрдгрдирд╛ рдХрд░реЗрдВред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
recall |
list
|
рдпрд╛рдж рдХрд░рдиреЗ рдХреА рд╡рдХреНрд░ред |
рдЖрд╡рд╢реНрдпрдХ |
precision |
list
|
рд╕рдЯреАрдХ рд╡рдХреНрд░ред |
рдЖрд╡рд╢реНрдпрдХ |
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
float
|
рдФрд╕рдд рд╕рдЯреАрдХрддрд╛ред |
ndarray
|
рд╕рдЯреАрдХ рд▓рд┐рдлрд╛рдлрд╛ рд╡рдХреНрд░ред |
ndarray
|
рд╢реБрд░реБрдЖрдд рдФрд░ рдЕрдВрдд рдореЗрдВ рдЬреЛрдбрд╝реЗ рдЧрдП рдкреНрд░рд╣рд░реА рдореВрд▓реНрдпреЛрдВ рдХреЗ рд╕рд╛рде рд╕рдВрд╢реЛрдзрд┐рдд рдпрд╛рдж рд╡рдХреНрд░ред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
ultralytics.utils.metrics.ap_per_class(tp, conf, pred_cls, target_cls, plot=False, on_plot=None, save_dir=Path(), names=(), eps=1e-16, prefix='')
рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди рдореВрд▓реНрдпрд╛рдВрдХрди рдХреЗ рд▓рд┐рдП рдкреНрд░рддрд┐ рд╡рд░реНрдЧ рдФрд╕рдд рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ рдХреА рдЧрдгрдирд╛ рдХрд░рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
tp |
ndarray
|
рдмрд╛рдЗрдирд░реА рд╕рд░рдгреА рдЗрдВрдЧрд┐рдд рдХрд░рддреА рд╣реИ рдХрд┐ рдкрддрд╛ рд▓рдЧрд╛рдирд╛ рд╕рд╣реА рд╣реИ (рд╕рддреНрдп) рдпрд╛ рдирд╣реАрдВ (рдЧрд▓рдд)ред |
рдЖрд╡рд╢реНрдпрдХ |
conf |
ndarray
|
рдкрддрд╛ рд▓рдЧрд╛рдиреЗ рдХреЗ рдЖрддреНрдорд╡рд┐рд╢реНрд╡рд╛рд╕ рд╕реНрдХреЛрд░ рдХреА рд╕рд░рдгреАред |
рдЖрд╡рд╢реНрдпрдХ |
pred_cls |
ndarray
|
рдкрддрд╛ рд▓рдЧрд╛рдиреЗ рдХреЗ рдЕрдиреБрдорд╛рдирд┐рдд рд╡рд░реНрдЧреЛрдВ рдХреА рд╕рд░рдгреАред |
рдЖрд╡рд╢реНрдпрдХ |
target_cls |
ndarray
|
рдкрддрд╛ рд▓рдЧрд╛рдиреЗ рдХреЗ рд╕рдЪреНрдЪреЗ рд╡рд░реНрдЧреЛрдВ рдХреА рд╕рд░рдгреАред |
рдЖрд╡рд╢реНрдпрдХ |
plot |
bool
|
рдкреАрдЖрд░ рдХрд░реНрд╡реНрд╕ рдкреНрд▓реЙрдЯ рдХрд░рдирд╛ рд╣реИ рдпрд╛ рдирд╣реАрдВред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рд░реВрдк рд╕реЗ рдЧрд▓рдд рд╣реИ. |
False
|
on_plot |
func
|
рдкреНрд▓реЙрдЯреНрд╕ рдкрде рдФрд░ рдбреЗрдЯрд╛ рдХреЛ рдкрд╛рд╕ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ рдХреЙрд▓рдмреИрдХ рдЬрдм рд╡реЗ рдкреНрд░рд╕реНрддреБрдд рдХрд┐рдП рдЬрд╛рддреЗ рд╣реИрдВред рдХреЛрдИ рдирд╣реАрдВ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
None
|
save_dir |
Path
|
рдкреАрдЖрд░ рдШрдЯрддрд╛ рдХреЛ рдмрдЪрд╛рдиреЗ рдХреЗ рд▓рд┐рдП рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ред рдПрдХ рдЦрд╛рд▓реА рдкрде рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
Path()
|
names |
tuple
|
рдкреАрдЖрд░ рдШрдЯрддрд╛ рд╕рд╛рдЬрд┐рд╢ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рд╡рд░реНрдЧ рдХреЗ рдирд╛рдореЛрдВ рдХрд╛ рдЯреБрдкрд▓ред рдПрдХ рдЦрд╛рд▓реА рдЯрдкрд▓ рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
()
|
eps |
float
|
рд╢реВрдиреНрдп рд╕реЗ рд╡рд┐рднрд╛рдЬрди рд╕реЗ рдмрдЪрдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ рдЫреЛрдЯрд╛ рдореВрд▓реНрдпред 1e-16 рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
1e-16
|
prefix |
str
|
рдкреНрд▓реЙрдЯ рдлрд╝рд╛рдЗрд▓реЛрдВ рдХреЛ рд╕рд╣реЗрдЬрдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ рдЙрдкрд╕рд░реНрдЧ рд╕реНрдЯреНрд░рд┐рдВрдЧред рдПрдХ рдЦрд╛рд▓реА рд╕реНрдЯреНрд░рд┐рдВрдЧ рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
''
|
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
tuple
|
рдЫрд╣ рд╕рд░рдгрд┐рдпреЛрдВ рдХрд╛ рдПрдХ рдЯрдкрд▓ рдФрд░ рдЕрджреНрд╡рд┐рддреАрдп рд╡рд░реНрдЧреЛрдВ рдХреА рдПрдХ рд╕рд░рдгреА, рдЬрд╣рд╛рдВ: tp (np.ndarray): рдкреНрд░рддреНрдпреЗрдХ рд╡рд░реНрдЧ рдХреЗ рд▓рд┐рдП рдЕрдзрд┐рдХрддрдо F1 рдореАрдЯреНрд░рд┐рдХ рджреНрд╡рд╛рд░рд╛ рджреА рдЧрдИ рд╕реАрдорд╛ рдкрд░ рд╕рд╣реА рд╕рдХрд╛рд░рд╛рддреНрдордХ рдЧрдгрдирд╛ рдХрд░рддрд╛ рд╣реИред рдЖрдХрд╛рд░: (рдПрдирд╕реА,)ред fp (np.ndarray): рдкреНрд░рддреНрдпреЗрдХ рд╡рд░реНрдЧ рдХреЗ рд▓рд┐рдП рдЕрдзрд┐рдХрддрдо F1 рдореАрдЯреНрд░рд┐рдХ рджреНрд╡рд╛рд░рд╛ рджреА рдЧрдИ рд╕реАрдорд╛ рдкрд░ рдЭреВрдареА рд╕рдХрд╛рд░рд╛рддреНрдордХ рдЧрдгрдирд╛ред рдЖрдХрд╛рд░: (рдПрдирд╕реА,)ред p (np.ndarray): рдкреНрд░рддреНрдпреЗрдХ рд╡рд░реНрдЧ рдХреЗ рд▓рд┐рдП рдЕрдзрд┐рдХрддрдо F1 рдореАрдЯреНрд░рд┐рдХ рджреНрд╡рд╛рд░рд╛ рджрд┐рдП рдЧрдП рдереНрд░реЗрд╢реЛрд▓реНрдб рдкрд░ рд╕рдЯреАрдХ рдорд╛рдиред рдЖрдХрд╛рд░: (nc,)ред r (np.ndarray): рдкреНрд░рддреНрдпреЗрдХ рд╡рд░реНрдЧ рдХреЗ рд▓рд┐рдП рдЕрдзрд┐рдХрддрдо F1 рдореАрдЯреНрд░рд┐рдХ рджреНрд╡рд╛рд░рд╛ рджрд┐рдП рдЧрдП рдереНрд░реЗрд╢реЛрд▓реНрдб рдкрд░ рдорд╛рдиреЛрдВ рдХреЛ рдпрд╛рдж рдХрд░реЗрдВред рдЖрдХрд╛рд░: (nc,)ред f1 (np.ndarray): рдкреНрд░рддреНрдпреЗрдХ рд╡рд░реНрдЧ рдХреЗ рд▓рд┐рдП рдЕрдзрд┐рдХрддрдо F1 рдореАрдЯреНрд░рд┐рдХ рджреНрд╡рд╛рд░рд╛ рджрд┐рдП рдЧрдП рдереНрд░реЗрд╢реЛрд▓реНрдб рдкрд░ F1-рд╕реНрдХреЛрд░ рдорд╛рдиред рдЖрдХрд╛рд░: (nc,)ред ap (np.ndarray): рдЕрд▓рдЧ-рдЕрд▓рдЧ IoU рдереНрд░реЗрд╕рд╣реЛрд▓реНрдб рдкрд░ рдкреНрд░рддреНрдпреЗрдХ рд╡рд░реНрдЧ рдХреЗ рд▓рд┐рдП рдФрд╕рдд рд╕рдЯреАрдХрддрд╛ред рдЖрдХрд╛рд░: (рдПрдирд╕реА, 10)ред unique_classes (np.ndarray): рдЕрджреНрд╡рд┐рддреАрдп рд╡рд░реНрдЧреЛрдВ рдХреА рдПрдХ рд╕рд░рдгреА рдЬрд┐рд╕рдореЗрдВ рдбреЗрдЯрд╛ рд╣реИред рдЖрдХрд╛рд░: (рдПрдирд╕реА,)ред p_curve (np.ndarray): рдкреНрд░рддреНрдпреЗрдХ рд╡рд░реНрдЧ рдХреЗ рд▓рд┐рдП рд╕рдЯреАрдХ рдШрдЯрддрд╛ред рдЖрдХрд╛рд░: (рдПрдирд╕реА, 1000)ред r_curve (np.ndarray): рдкреНрд░рддреНрдпреЗрдХ рд╡рд░реНрдЧ рдХреЗ рд▓рд┐рдП рдХрд░реНрд╡реНрд╕ рд░рд┐рдХреЙрд▓ рдХрд░реЗрдВред рдЖрдХрд╛рд░: (рдПрдирд╕реА, 1000)ред f1_curve (np.ndarray): рдкреНрд░рддреНрдпреЗрдХ рд╡рд░реНрдЧ рдХреЗ рд▓рд┐рдП F1-рд╕реНрдХреЛрд░ рдШрдЯрддрд╛ред рдЖрдХрд╛рд░: (рдПрдирд╕реА, 1000)ред x (np.ndarray): рд╡рдХреНрд░реЛрдВ рдХреЗ рд▓рд┐рдП X-рдЕрдХреНрд╖ рдорд╛рдиред рдЖрдХрд╛рд░: (1000,)ред prec_values: рдкреНрд░рддреНрдпреЗрдХ рд╡рд░реНрдЧ рдХреЗ рд▓рд┐рдП mAP@0.5 рдкрд░ рд╕рдЯреАрдХ рдорд╛рдиред рдЖрдХрд╛рд░: (рдПрдирд╕реА, 1000)ред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/utils/metrics.py
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 556 557 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 |
|
2023-11-12 рдмрдирд╛рдпрд╛ рдЧрдпрд╛, рдЕрдкрдбреЗрдЯ рдХрд┐рдпрд╛ рдЧрдпрд╛ 2024-05-08
рд▓реЗрдЦрдХ: рдмреБрд░рд╣рд╛рди-рдХреНрдпреВ (1), рдЧреНрд▓реЗрди-рдЬреЛрдЪрд░ (5), рд▓рд╛рдлрд┐рдВрдЧ-рдХреНрдпреВ (1)