рдХреЗ рд▓рд┐рдП рд╕рдВрджрд░реНрдн ultralytics/data/utils.py
рдиреЛрдЯ
рдпрд╣ рдлрд╝рд╛рдЗрд▓ рдпрд╣рд╛рдБ рдЙрдкрд▓рдмреНрдз рд╣реИ https://github.com/ultralytics/ultralytics/рдмреВрдБрдж/рдореБрдЦреНрдп/ultralytics/data/utils.py рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░реЗрдВред рдпрджрд┐ рдЖрдк рдХреЛрдИ рд╕рдорд╕реНрдпрд╛ рджреЗрдЦрддреЗ рд╣реИрдВ рддреЛ рдХреГрдкрдпрд╛ рдкреБрд▓ рдЕрдиреБрд░реЛрдз рдХрд╛ рдпреЛрдЧрджрд╛рди рдХрд░рдХреЗ рдЗрд╕реЗ рдареАрдХ рдХрд░рдиреЗ рдореЗрдВ рдорджрдж рдХрд░реЗрдВ ЁЯЫая╕Пред ЁЯЩП рдзрдиреНрдпрд╡рд╛рдж !
ultralytics.data.utils.HUBDatasetStats
HUB рдбреЗрдЯрд╛рд╕реЗрдЯ JSON рдФрд░ рдЬрдирд░реЗрдЯ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ рдХреНрд▓рд╛рд╕ -hub
рдбреЗрдЯрд╛рд╕реЗрдЯ рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
path |
str
|
data.yaml рдпрд╛ data.zip рдХрд╛ рдкрде (data.zip рдЕрдВрджрд░ data.yaml рдХреЗ рд╕рд╛рде)ред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 'coco8.yaml' рд╣реИред |
'coco8.yaml'
|
task |
str
|
рдбреЗрдЯрд╛рд╕реЗрдЯ рдХрд╛рд░реНрдпред рд╡рд┐рдХрд▓реНрдк 'рдкрддрд╛ рд▓рдЧрд╛рдПрдВ', 'рд╕реЗрдЧрдореЗрдВрдЯ', 'рдкреЛрдЬрд╝', 'рд╡рд░реНрдЧреАрдХреГрдд' рд╣реИрдВред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 'рдкрддрд╛ рд▓рдЧрд╛рдирд╛' рд╣реИред |
'detect'
|
autodownload |
bool
|
рд╕реНрдерд╛рдиреАрдп рд░реВрдк рд╕реЗ рдирд╣реАрдВ рдорд┐рд▓рдиреЗ рдкрд░ рдбреЗрдЯрд╛рд╕реЗрдЯ рдбрд╛рдЙрдирд▓реЛрдб рдХрд░рдиреЗ рдХрд╛ рдкреНрд░рдпрд╛рд╕ рдХрд░реЗрдВред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рдЧрд╝рд▓рдд рд╣реИ. |
False
|
рдЙрджрд╛рд╣рд░рдг
https://github.com/ рд╕реЗ * .zip рдлрд╝рд╛рдЗрд▓реЗрдВ рдбрд╛рдЙрдирд▓реЛрдб рдХрд░реЗрдВultralytics/рд╣рдм/рдЯреНрд░реА/рдореБрдЦреНрдп/example_datasets рдпрд╛рдиреА https://github.com/ultralytics/hub/raw/main/example_datasets/coco8.zip coco8.zip рдХреЗ рд▓рд┐рдПред
from ultralytics.data.utils import HUBDatasetStats
stats = HUBDatasetStats('path/to/coco8.zip', task='detect') # detect dataset
stats = HUBDatasetStats('path/to/coco8-seg.zip', task='segment') # segment dataset
stats = HUBDatasetStats('path/to/coco8-pose.zip', task='pose') # pose dataset
stats = HUBDatasetStats('path/to/dota8.zip', task='obb') # OBB dataset
stats = HUBDatasetStats('path/to/imagenet10.zip', task='classify') # classification dataset
stats.get_json(save=True)
stats.process_images()
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/utils.py
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 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 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 |
|
__init__(path='coco8.yaml', task='detect', autodownload=False)
рдХреНрд▓рд╛рд╕ рдЗрдирд┐рд╢рд┐рдпрд▓рд╛рдЗрдЬрд╝ рдХрд░реЗрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/utils.py
get_json(save=False, verbose=False)
рдХреЗ рд▓рд┐рдП рдбреЗрдЯрд╛рд╕реЗрдЯ JSON рд▓реМрдЯрд╛рдПрдВ Ultralytics рдЪрдХреНрд░рдирд╛рднрд┐ред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/utils.py
process_images()
рдХреЗ рд▓рд┐рдП рдЫрд╡рд┐рдпреЛрдВ рдХреЛ рд╕рдВрдкреАрдбрд╝рд┐рдд рдХрд░реЗрдВ Ultralytics рдЪрдХреНрд░рдирд╛рднрд┐ред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/utils.py
ultralytics.data.utils.img2label_paths(img_paths)
рд▓реЗрдмрд▓ рдкрде рдХреЛ рдЫрд╡рд┐ рдкрде рдХреЗ рдПрдХ рдлрд╝рдВрдХреНрд╢рди рдХреЗ рд░реВрдк рдореЗрдВ рдкрд░рд┐рднрд╛рд╖рд┐рдд рдХрд░реЗрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/utils.py
ultralytics.data.utils.get_hash(paths)
рдкрде (рдлрд╝рд╛рдЗрд▓реЛрдВ рдпрд╛ dirs) рдХреА рд╕реВрдЪреА рдХрд╛ рдПрдХрд▓ рд╣реИрд╢ рдорд╛рди рджреЗрддрд╛ рд╣реИред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/utils.py
ultralytics.data.utils.exif_size(img)
exif-рд╕рд╣реА PIL рдЖрдХрд╛рд░ рджреЗрддрд╛ рд╣реИ.
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/utils.py
ultralytics.data.utils.verify_image(args)
рдПрдХ рдЫрд╡рд┐ рд╕рддреНрдпрд╛рдкрд┐рдд рдХрд░реЗрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/utils.py
ultralytics.data.utils.verify_image_label(args)
рдПрдХ рдЫрд╡рд┐-рд▓реЗрдмрд▓ рдЬреЛрдбрд╝реА рд╕рддреНрдпрд╛рдкрд┐рдд рдХрд░реЗрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/utils.py
ultralytics.data.utils.polygon2mask(imgsz, polygons, color=1, downsample_ratio=1)
рдмрд╣реБрднреБрдЬреЛрдВ рдХреА рд╕реВрдЪреА рдХреЛ рдирд┐рд░реНрджрд┐рд╖реНрдЯ рдЫрд╡рд┐ рдЖрдХрд╛рд░ рдХреЗ рдмрд╛рдЗрдирд░реА рдорд╛рд╕реНрдХ рдореЗрдВ рдХрдирд╡рд░реНрдЯ рдХрд░реЗрдВред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
imgsz |
tuple
|
рдЫрд╡рд┐ рдХрд╛ рдЖрдХрд╛рд░ (рдКрдВрдЪрд╛рдИ, рдЪреМрдбрд╝рд╛рдИ) рдХреЗ рд░реВрдк рдореЗрдВред |
рдЖрд╡рд╢реНрдпрдХ |
polygons |
list[ndarray]
|
рдмрд╣реБрднреБрдЬреЛрдВ рдХреА рдПрдХ рд╕реВрдЪреАред рдкреНрд░рддреНрдпреЗрдХ рдмрд╣реБрднреБрдЬ рдЖрдХрд╛рд░ [N, M] рдХреЗ рд╕рд╛рде рдПрдХ рд╕рд░рдгреА рд╣реИ, рдЬрд╣рд╛рдБ N рдмрд╣реБрднреБрдЬреЛрдВ рдХреА рд╕рдВрдЦреНрдпрд╛ рд╣реИ, рдФрд░ M рдмрд┐рдВрджреБрдУрдВ рдХреА рд╕рдВрдЦреНрдпрд╛ рд╣реИ рдЬреИрд╕реЗ рдХрд┐ M% 2 = 0ред |
рдЖрд╡рд╢реНрдпрдХ |
color |
int
|
рдореБрдЦреМрдЯрд╛ рдкрд░ рдмрд╣реБрднреБрдЬ рднрд░рдиреЗ рдХреЗ рд▓рд┐рдП рд░рдВрдЧ рдорд╛рдиред 1 рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
1
|
downsample_ratio |
int
|
рдХрд╛рд░рдХ рдЬрд┐рд╕рдХреЗ рджреНрд╡рд╛рд░рд╛ рдореБрдЦреМрдЯрд╛ downsample рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП. 1 рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
1
|
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
ndarray
|
рдирд┐рд░реНрджрд┐рд╖реНрдЯ рдЫрд╡рд┐ рдЖрдХрд╛рд░ рдХрд╛ рдПрдХ рдмрд╛рдЗрдирд░реА рдорд╛рд╕реНрдХ рдЬрд┐рд╕рдореЗрдВ рдмрд╣реБрднреБрдЬ рднрд░реЗ рд╣реБрдП рд╣реИрдВред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/utils.py
ultralytics.data.utils.polygons2masks(imgsz, polygons, color, downsample_ratio=1)
рдмрд╣реБрднреБрдЬреЛрдВ рдХреА рдПрдХ рд╕реВрдЪреА рдХреЛ рдирд┐рд░реНрджрд┐рд╖реНрдЯ рдЫрд╡рд┐ рдЖрдХрд╛рд░ рдХреЗ рдмрд╛рдЗрдирд░реА рдорд╛рд╕реНрдХ рдХреЗ рдПрдХ рд╕реЗрдЯ рдореЗрдВ рдХрдирд╡рд░реНрдЯ рдХрд░реЗрдВред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
imgsz |
tuple
|
рдЫрд╡рд┐ рдХрд╛ рдЖрдХрд╛рд░ (рдКрдВрдЪрд╛рдИ, рдЪреМрдбрд╝рд╛рдИ) рдХреЗ рд░реВрдк рдореЗрдВред |
рдЖрд╡рд╢реНрдпрдХ |
polygons |
list[ndarray]
|
рдмрд╣реБрднреБрдЬреЛрдВ рдХреА рдПрдХ рд╕реВрдЪреАред рдкреНрд░рддреНрдпреЗрдХ рдмрд╣реБрднреБрдЬ рдЖрдХрд╛рд░ [N, M] рдХреЗ рд╕рд╛рде рдПрдХ рд╕рд░рдгреА рд╣реИ, рдЬрд╣рд╛рдБ N рдмрд╣реБрднреБрдЬреЛрдВ рдХреА рд╕рдВрдЦреНрдпрд╛ рд╣реИ, рдФрд░ M рдмрд┐рдВрджреБрдУрдВ рдХреА рд╕рдВрдЦреНрдпрд╛ рд╣реИ рдЬреИрд╕реЗ рдХрд┐ M% 2 = 0ред |
рдЖрд╡рд╢реНрдпрдХ |
color |
int
|
рдорд╛рд╕реНрдХ рдкрд░ рдмрд╣реБрднреБрдЬ рднрд░рдиреЗ рдХреЗ рд▓рд┐рдП рд░рдВрдЧ рдореВрд▓реНрдпред |
рдЖрд╡рд╢реНрдпрдХ |
downsample_ratio |
int
|
рдХрд╛рд░рдХ рдЬрд┐рд╕рдХреЗ рджреНрд╡рд╛рд░рд╛ рдкреНрд░рддреНрдпреЗрдХ рдореБрдЦреМрдЯрд╛ downsample рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП. 1 рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
1
|
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
ndarray
|
рднрд░реЗ рдЧрдП рдмрд╣реБрднреБрдЬреЛрдВ рдХреЗ рд╕рд╛рде рдирд┐рд░реНрджрд┐рд╖реНрдЯ рдЫрд╡рд┐ рдЖрдХрд╛рд░ рдХреЗ рдмрд╛рдЗрдирд░реА рдорд╛рд╕реНрдХ рдХрд╛ рдПрдХ рд╕реЗрдЯред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/utils.py
ultralytics.data.utils.polygons2masks_overlap(imgsz, segments, downsample_ratio=1)
рдПрдХ (640, 640) рдУрд╡рд░рд▓реИрдк рдорд╛рд╕реНрдХ рд▓реМрдЯрд╛рдПрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/utils.py
ultralytics.data.utils.find_dataset_yaml(path)
рдбрд┐рдЯреЗрдХреНрдЯ, рд╕реЗрдЧрдореЗрдВрдЯ рдпрд╛ рдкреЛрдЬрд╝ рдбреЗрдЯрд╛рд╕реЗрдЯ рд╕реЗ рдЬреБрдбрд╝реА YAML рдлрд╝рд╛рдЗрд▓ рдвреВрдВрдвреЗрдВ рдФрд░ рд▓реМрдЯрд╛рдПрдВред
рдпрд╣ рдлрд╝рдВрдХреНрд╢рди рдкрд╣рд▓реЗ рдкреНрд░рджрд╛рди рдХреА рдЧрдИ рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рдХреЗ рд░реВрдЯ рд╕реНрддрд░ рдкрд░ рдПрдХ YAML рдлрд╝рд╛рдЗрд▓ рдХреА рдЦреЛрдЬ рдХрд░рддрд╛ рд╣реИ, рдФрд░ рдпрджрд┐ рдирд╣реАрдВ рдорд┐рд▓рд╛, рддреЛ рдпрд╣ рдПрдХ рдкреБрдирд░рд╛рд╡рд░реНрддреА рдЦреЛрдЬ рдХрд░рддрд╛ рд╣реИред рдпрд╣ YAML рдлрд╝рд╛рдЗрд▓реЛрдВ рдХреЛ рдкреНрд░рд╛рдердорд┐рдХрддрд╛ рджреЗрддрд╛ рд╣реИ рдЬрд┐рдирдореЗрдВ рдкреНрд░рджрд╛рди рдХрд┐рдП рдЧрдП рдкрде рдХреЗ рд╕рдорд╛рди рд╕реНрдЯреЗрдо рд╣реЛрддрд╛ рд╣реИред рдПрдХ рдЕрднрд┐рдХрдерди рддреНрд░реБрдЯрд┐ рдпрджрд┐ рдХреЛрдИ YAML рдлрд╝рд╛рдЗрд▓ рдирд╣реАрдВ рдорд┐рд▓рддреА рд╣реИ рдпрд╛ рдпрджрд┐ рдПрдХрд╛рдзрд┐рдХ YAML рдлрд╝рд╛рдЗрд▓реЗрдВ рдорд┐рд▓рддреА рд╣реИрдВ, рддреЛ рдЙрдард╛рдпрд╛ рдЬрд╛рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
path |
Path
|
YAML рдлрд╝рд╛рдЗрд▓ рдХреЗ рд▓рд┐рдП рдЦреЛрдЬ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рдкрдеред |
рдЖрд╡рд╢реНрдпрдХ |
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
Path
|
рдкрд╛рдпрд╛ YAML рдлрд╝рд╛рдЗрд▓ рдХрд╛ рдкрдеред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/utils.py
ultralytics.data.utils.check_det_dataset(dataset, autodownload=True)
рдпрджрд┐ рд╕реНрдерд╛рдиреАрдп рд░реВрдк рд╕реЗ рдирд╣реАрдВ рдорд┐рд▓рддрд╛ рд╣реИ рддреЛ рдбреЗрдЯрд╛рд╕реЗрдЯ рдбрд╛рдЙрдирд▓реЛрдб рдХрд░реЗрдВ, рд╕рддреНрдпрд╛рдкрд┐рдд рдХрд░реЗрдВ рдФрд░/рдпрд╛ рдЕрдирдЬрд╝рд┐рдк рдХрд░реЗрдВред
рдпрд╣ рдлрд╝рдВрдХреНрд╢рди рдПрдХ рдирд┐рд░реНрджрд┐рд╖реНрдЯ рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреА рдЙрдкрд▓рдмреНрдзрддрд╛ рдХреА рдЬрд╛рдВрдЪ рдХрд░рддрд╛ рд╣реИ, рдФрд░ рдпрджрд┐ рдирд╣реАрдВ рдорд┐рд▓рддрд╛ рд╣реИ, рддреЛ рдЗрд╕рдореЗрдВ рдбрд╛рдЙрдирд▓реЛрдб рдХрд░рдиреЗ рдХрд╛ рд╡рд┐рдХрд▓реНрдк рд╣реЛрддрд╛ рд╣реИ рдФрд░ рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреЛ рдЕрдирдЬрд╝рд┐рдк рдХрд░реЗрдВред рдпрд╣ рддрдм рд╡рд╛рдИрдПрдПрдордПрд▓ рдбреЗрдЯрд╛ рдХреЗ рд╕рд╛рде рдкрдврд╝рддрд╛ рд╣реИ рдФрд░ рдкрд╛рд░реНрд╕ рдХрд░рддрд╛ рд╣реИ, рдпрд╣ рд╕реБрдирд┐рд╢реНрдЪрд┐рдд рдХрд░рддрд╛ рд╣реИ рдХрд┐ рдкреНрд░рдореБрдЦ рдЖрд╡рд╢реНрдпрдХрддрд╛рдУрдВ рдХреЛ рдкреВрд░рд╛ рдХрд┐рдпрд╛ рдЬрд╛рдП рдФрд░ рдпрд╣ рднреА рдбреЗрдЯрд╛рд╕реЗрдЯ рд╕реЗ рд╕рдВрдмрдВрдзрд┐рдд рдкрдереЛрдВ рдХрд╛ рд╕рдорд╛рдзрд╛рди рдХрд░рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
dataset |
str
|
рдбреЗрдЯрд╛рд╕реЗрдЯ рдпрд╛ рдбреЗрдЯрд╛рд╕реЗрдЯ рдбрд┐рд╕реНрдХреНрд░рд┐рдкреНрдЯрд░ (рдЬреИрд╕реЗ YAML рдлрд╝рд╛рдЗрд▓) рдХрд╛ рдкрдеред |
рдЖрд╡рд╢реНрдпрдХ |
autodownload |
bool
|
рдпрджрд┐ рдирд╣реАрдВ рдорд┐рд▓рд╛ рддреЛ рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреЛ рд╕реНрд╡рдЪрд╛рд▓рд┐рдд рд░реВрдк рд╕реЗ рдбрд╛рдЙрдирд▓реЛрдб рдХрд░рдирд╛ рд╣реИ рдпрд╛ рдирд╣реАрдВред рд╕рд╣реА рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
True
|
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
dict
|
рдкрд╛рд░реНрд╕ рдХрд┐рдП рдЧрдП рдбреЗрдЯрд╛рд╕реЗрдЯ рдЬрд╛рдирдХрд╛рд░реА рдФрд░ рдкрдеред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/utils.py
253 254 255 256 257 258 259 260 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 |
|
ultralytics.data.utils.check_cls_dataset(dataset, split='')
рдЗрдореЗрдЬрдиреЗрдЯ рдЬреИрд╕реЗ рд╡рд░реНрдЧреАрдХрд░рдг рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреА рдЬрд╛рдБрдЪ рдХрд░рддрд╛ рд╣реИред
рдпрд╣ рдлрд╝рдВрдХреНрд╢рди рдПрдХ рд╕реНрд╡реАрдХрд╛рд░ рдХрд░рддрд╛ рд╣реИ dataset
рдирд╛рдо рдФрд░ рд╕рдВрдмрдВрдзрд┐рдд рдбреЗрдЯрд╛рд╕реЗрдЯ рдЬрд╛рдирдХрд╛рд░реА рдХреЛ рдкреБрдирдГ рдкреНрд░рд╛рдкреНрдд рдХрд░рдиреЗ рдХрд╛ рдкреНрд░рдпрд╛рд╕ред
рдпрджрд┐ рдбреЗрдЯрд╛рд╕реЗрдЯ рд╕реНрдерд╛рдиреАрдп рд░реВрдк рд╕реЗ рдирд╣реАрдВ рдорд┐рд▓рддрд╛ рд╣реИ, рддреЛ рдпрд╣ рдЗрдВрдЯрд░рдиреЗрдЯ рд╕реЗ рдбреЗрдЯрд╛рд╕реЗрдЯ рдбрд╛рдЙрдирд▓реЛрдб рдХрд░рдиреЗ рдФрд░ рдЗрд╕реЗ рд╕реНрдерд╛рдиреАрдп рд░реВрдк рд╕реЗ рд╕рд╣реЗрдЬрдиреЗ рдХрд╛ рдкреНрд░рдпрд╛рд╕ рдХрд░рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
dataset |
str | Path
|
рдбреЗрдЯрд╛рд╕реЗрдЯ рдХрд╛ рдирд╛рдо. |
рдЖрд╡рд╢реНрдпрдХ |
split |
str
|
рдбреЗрдЯрд╛рд╕реЗрдЯ рдХрд╛ рд╡рд┐рднрд╛рдЬрдиред рдпрд╛ рддреЛ 'рд╡реИрд▓', 'рдЯреЗрд╕реНрдЯ', рдпрд╛ ''ред '' рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
''
|
рджреЗрддрд╛:
рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|
dict
|
рдПрдХ рд╢рдмреНрджрдХреЛрд╢ рдЬрд┐рд╕рдореЗрдВ рдирд┐рдореНрдирд▓рд┐рдЦрд┐рдд рдХреБрдВрдЬрд┐рдпрд╛рдБ рд╣реЛрддреА рд╣реИрдВ: - 'рдЯреНрд░реЗрди' (рдкрде): рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреЗ рдкреНрд░рд╢рд┐рдХреНрд╖рдг рд╕реЗрдЯ рд╡рд╛рд▓реЗ рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рдкрдеред - 'val' (рдкрде): рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреЗ рд╕рддреНрдпрд╛рдкрди рд╕реЗрдЯ рд╡рд╛рд▓реЗ рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рдкрдеред - 'рдкрд░реАрдХреНрд╖рдг' (рдкрде): рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреЗ рдкрд░реАрдХреНрд╖рдг рд╕реЗрдЯ рд╡рд╛рд▓реЗ рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рдкрдеред - 'nc' (int): рдбреЗрдЯрд╛рд╕реЗрдЯ рдореЗрдВ рдХрдХреНрд╖рд╛рдУрдВ рдХреА рд╕рдВрдЦреНрдпрд╛ред - 'рдирд╛рдо' (рдбрд┐рдХреНрдЯ): рдбреЗрдЯрд╛рд╕реЗрдЯ рдореЗрдВ рд╡рд░реНрдЧ рдирд╛рдореЛрдВ рдХрд╛ рдПрдХ рд╢рдмреНрджрдХреЛрд╢ред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/utils.py
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 |
|
ultralytics.data.utils.compress_one_image(f, f_new=None, max_dim=1920, quality=50)
рдХрд┐рд╕реА рдПрдХрд▓ рдЫрд╡рд┐ рдлрд╝рд╛рдЗрд▓ рдХреЛ рдЙрд╕рдХреЗ рдкрд╣рд▓реВ рдЕрдиреБрдкрд╛рдд рдФрд░ рдЧреБрдгрд╡рддреНрддрд╛ рдХреЛ рд╕рдВрд░рдХреНрд╖рд┐рдд рдХрд░рддреЗ рд╣реБрдП рдХрдо рдЖрдХрд╛рд░ рдореЗрдВ рд╕рдВрдкреАрдбрд╝рд┐рдд рдХрд░рддрд╛ рд╣реИ Python рдЗрдореЗрдЬрд┐рдВрдЧ рд▓рд╛рдЗрдмреНрд░реЗрд░реА (рдкреАрдЖрдИрдПрд▓) рдпрд╛ рдУрдкрдирд╕реАрд╡реА рд▓рд╛рдЗрдмреНрд░реЗрд░реАред рдпрджрд┐ рдЗрдирдкреБрдЯ рдЫрд╡рд┐ рдЕрдзрд┐рдХрддрдо рдЖрдпрд╛рдо рд╕реЗ рдЫреЛрдЯреА рд╣реИ, рддреЛ рдпрд╣ рдирд╣реАрдВ рд╣реЛрдЧреА рдЖрдХрд╛рд░ рдмрджрд▓рд╛ред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
f |
str
|
рдЗрдирдкреБрдЯ рдЫрд╡рд┐ рдлрд╝рд╛рдЗрд▓ рдХрд╛ рдкрдеред |
рдЖрд╡рд╢реНрдпрдХ |
f_new |
str
|
рдЖрдЙрдЯрдкреБрдЯ рдЫрд╡рд┐ рдлрд╝рд╛рдЗрд▓ рдХрд╛ рдкрдеред рдпрджрд┐ рдирд┐рд░реНрджрд┐рд╖реНрдЯ рдирд╣реАрдВ рд╣реИ, рддреЛ рдЗрдирдкреБрдЯ рдлрд╝рд╛рдЗрд▓ рдЕрдзрд┐рд▓реЗрдЦрд┐рдд рд╣реЛ рдЬрд╛рдПрдЧреАред |
None
|
max_dim |
int
|
рдЖрдЙрдЯрдкреБрдЯ рдЫрд╡рд┐ рдХрд╛ рдЕрдзрд┐рдХрддрдо рдЖрдпрд╛рдо (рдЪреМрдбрд╝рд╛рдИ рдпрд╛ рдКрдВрдЪрд╛рдИ)ред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 1920 рдкрд┐рдХреНрд╕реЗрд▓ рд╣реИред |
1920
|
quality |
int
|
рдкреНрд░рддрд┐рд╢рдд рдХреЗ рд░реВрдк рдореЗрдВ рдЫрд╡рд┐ рд╕рдВрдкреАрдбрд╝рди рдЧреБрдгрд╡рддреНрддрд╛ред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 50% рд╣реИред |
50
|
рдЙрджрд╛рд╣рд░рдг
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/utils.py
ultralytics.data.utils.autosplit(path=DATASETS_DIR / 'coco8/images', weights=(0.9, 0.1, 0.0), annotated_only=False)
рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреЛ рдЯреНрд░реЗрди/рд╡реИрд▓/рдЯреЗрд╕реНрдЯ рд╕реНрдкреНрд▓рд┐рдЯ рдореЗрдВ рд╕реНрд╡рдЪрд╛рд▓рд┐рдд рд░реВрдк рд╕реЗ рд╡рд┐рднрд╛рдЬрд┐рдд рдХрд░реЗрдВ рдФрд░ рдкрд░рд┐рдгрд╛рдореА рд╕реНрдкреНрд▓рд┐рдЯ рдХреЛ autosplit_*.txt рдлрд╝рд╛рдЗрд▓реЛрдВ рдореЗрдВ рд╕рд╣реЗрдЬреЗрдВред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
path |
Path
|
рдЫрд╡рд┐рдпреЛрдВ рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рдХреЗ рд▓рд┐рдП рдкрдеред 'coco8/images' DATASETS_DIR рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдбрд┐рдлрд╝реЙрд▓реНрдЯред |
DATASETS_DIR / 'coco8/images'
|
weights |
list | tuple
|
рдЯреНрд░реЗрди, рд╕рддреНрдпрд╛рдкрди, рдФрд░ рдкрд░реАрдХреНрд╖рдг рд╡рд┐рднрд╛рдЬрди рдЕрдВрд╢ред рдбрд┐рдлрд╝реЙрд▓реНрдЯ (0.9, 0.1, 0.0)ред |
(0.9, 0.1, 0.0)
|
annotated_only |
bool
|
рдЕрдЧрд░ рд╕рд╣реА рд╣реИ, рддреЛ рд╕рд┐рд░реНрдлрд╝ рдЙрдирд╕реЗ рдЬреБрдбрд╝реА txt рдлрд╝рд╛рдЗрд▓ рд╡рд╛рд▓реА рдЗрдореЗрдЬ рдХрд╛ рдЗрд╕реНрддреЗрдорд╛рд▓ рдХрд┐рдпрд╛ рдЬрд╛рддрд╛ рд╣реИ. рдбрд┐рдлрд╝реЙрд▓реНрдЯ рд░реВрдк рд╕реЗ рдЧрд▓рдд рд╣реИ. |
False
|
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/utils.py
ultralytics.data.utils.load_dataset_cache_file(path)
рдПрдХ рд▓реЛрдб рдХрд░реЗрдВ Ultralytics *.cache рдкрде рд╕реЗ рд╢рдмреНрджрдХреЛрд╢ред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/utils.py
ultralytics.data.utils.save_dataset_cache_file(prefix, path, x, version)
рдПрдХ рд╕рд╣реЗрдЬреЗрдВ Ultralytics рдбреЗрдЯрд╛рд╕реЗрдЯ *.cache рдбрд┐рдХреНрд╢рдирд░реА x рдЯреВ рдкрд╛рдеред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/utils.py
2023-11-12 рдмрдирд╛рдпрд╛ рдЧрдпрд╛, рдЕрдкрдбреЗрдЯ рдХрд┐рдпрд╛ рдЧрдпрд╛ 2024-05-08
рд▓реЗрдЦрдХ: рдмреБрд░рд╣рд╛рди-рдХреНрдпреВ (1), рд▓рд╛рдлрд┐рдВрдЧ-рдХреНрдпреВ (1), рдЧреНрд▓реЗрди-рдЬреЛрдЪрд░ (3)