рдХреЗ рд▓рд┐рдП рд╕рдВрджрд░реНрдн ultralytics/data/build.py
рдиреЛрдЯ
рдпрд╣ рдлрд╝рд╛рдЗрд▓ рдпрд╣рд╛рдБ рдЙрдкрд▓рдмреНрдз рд╣реИ https://github.com/ultralytics/ultralytics/рдмреВрдБрдж/рдореБрдЦреНрдп/ultralytics/data/build.py рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░реЗрдВред рдпрджрд┐ рдЖрдк рдХреЛрдИ рд╕рдорд╕реНрдпрд╛ рджреЗрдЦрддреЗ рд╣реИрдВ рддреЛ рдХреГрдкрдпрд╛ рдкреБрд▓ рдЕрдиреБрд░реЛрдз рдХрд╛ рдпреЛрдЧрджрд╛рди рдХрд░рдХреЗ рдЗрд╕реЗ рдареАрдХ рдХрд░рдиреЗ рдореЗрдВ рдорджрдж рдХрд░реЗрдВ ЁЯЫая╕Пред ЁЯЩП рдзрдиреНрдпрд╡рд╛рдж !
ultralytics.data.build.InfiniteDataLoader
рдХрд╛ рд░реВрдк: DataLoader
рдбреЗрдЯрд╛рд▓реЛрдбрд░ рдЬреЛ рд╢реНрд░рдорд┐рдХреЛрдВ рдХрд╛ рдкреБрди: рдЙрдкрдпреЛрдЧ рдХрд░рддрд╛ рд╣реИред
рд╡реЗрдирд┐рд▓рд╛ рдбреЗрдЯрд╛рд▓реЛрдбрд░ рдХреЗ рд╕рдорд╛рди рд╕рд┐рдВрдЯреИрдХреНрд╕ рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░рддрд╛ рд╣реИред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/build.py
__init__(*args, **kwargs)
рдбреЗрдЯрд╛рд▓реЛрдбрд░ рдЬреЛ рд╢реНрд░рдорд┐рдХреЛрдВ рдХреЛ рдЕрд╕реАрдо рд░реВрдк рд╕реЗ рд░реАрд╕рд╛рдпрдХрд▓ рдХрд░рддрд╛ рд╣реИ, рдбреЗрдЯрд╛рд▓реЛрдбрд░ рд╕реЗ рд╡рд┐рд░рд╛рд╕рдд рдореЗрдВ рдорд┐рд▓рд╛ рд╣реИред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/build.py
__iter__()
рдПрдХ рдирдореВрдирд╛ рдмрдирд╛рддрд╛ рд╣реИ рдЬреЛ рдЕрдирд┐рд╢реНрдЪрд┐рдд рдХрд╛рд▓ рддрдХ рджреЛрд╣рд░рд╛рддрд╛ рд╣реИред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/build.py
__len__()
reset()
рдЗрдЯрд░реЗрдЯрд░ рд░реАрд╕реЗрдЯ рдХрд░реЗрдВред
рдпрд╣ рддрдм рдЙрдкрдпреЛрдЧреА рд╣реЛрддрд╛ рд╣реИ рдЬрдм рд╣рдо рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдХреЗ рджреМрд░рд╛рди рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреА рд╕реЗрдЯрд┐рдВрдЧреНрд╕ рдХреЛ рд╕рдВрд╢реЛрдзрд┐рдд рдХрд░рдирд╛ рдЪрд╛рд╣рддреЗ рд╣реИрдВред
ultralytics.data.build._RepeatSampler
рдирдореВрдирд╛ рдЬреЛ рд╣рдореЗрд╢рд╛ рдХреЗ рд▓рд┐рдП рджреЛрд╣рд░рд╛рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
sampler |
sampler
|
рджреЛрд╣рд░рд╛рдиреЗ рдХреЗ рд▓рд┐рдП рдирдореВрдирд╛ред |
рдЖрд╡рд╢реНрдпрдХ |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/build.py
__init__(sampler)
рдПрдХ рдСрдмреНрдЬреЗрдХреНрдЯ рдХреЛ рдЗрдирд┐рд╢рд┐рдпрд▓рд╛рдЗрдЬрд╝ рдХрд░рддрд╛ рд╣реИ рдЬреЛ рдХрд┐рд╕реА рджрд┐рдП рдЧрдП рд╕реИрдВрдкрд▓рд░ рдХреЛ рдЕрдирд┐рд╢реНрдЪрд┐рдд рдХрд╛рд▓ рддрдХ рджреЛрд╣рд░рд╛рддрд╛ рд╣реИред
__iter__()
'рдирдореВрдирд╛' рдкрд░ рдкреБрдирд░рд╛рд╡реГрддреНрдд рдХрд░рддрд╛ рд╣реИ рдФрд░ рдЗрд╕рдХреА рд╕рд╛рдордЧреНрд░реА рдЙрддреНрдкрдиреНрди рдХрд░рддрд╛ рд╣реИред
ultralytics.data.build.seed_worker(worker_id)
рдбреЗрдЯрд╛рд▓реЛрдбрд░ рд╡рд░реНрдХрд░ рд╕реАрдб https:// рд╕реЗрдЯ рдХрд░реЗрдВpytorch.org/docs/stable/notes/randomness.html#dataloader рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░реЗрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/build.py
ultralytics.data.build.build_yolo_dataset(cfg, img_path, batch, data, mode='train', rect=False, stride=32, multi_modal=False)
рдирд┐рд░реНрдорд╛рдг рдХрд░ YOLO рдбреЗрдЯрд╛рд╕реЗрдЯред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/build.py
ultralytics.data.build.build_grounding(cfg, img_path, json_file, batch, mode='train', rect=False, stride=32)
рдирд┐рд░реНрдорд╛рдг рдХрд░ YOLO рдбреЗрдЯрд╛рд╕реЗрдЯред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/build.py
ultralytics.data.build.build_dataloader(dataset, batch, workers, shuffle=True, rank=-1)
рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдпрд╛ рд╕рддреНрдпрд╛рдкрди рд╕реЗрдЯ рдХреЗ рд▓рд┐рдП рдПрдХ InfiniteDataLoader рдпрд╛ DataLoader рд▓реМрдЯрд╛рдПрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/build.py
ultralytics.data.build.check_source(source)
рд╕реНрд░реЛрдд рдкреНрд░рдХрд╛рд░ рдХреА рдЬрд╛рдБрдЪ рдХрд░реЗрдВ рдФрд░ рд╕рдВрдмрдВрдзрд┐рдд рдзреНрд╡рдЬ рдорд╛рди рд▓реМрдЯрд╛рдПрдБред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/build.py
ultralytics.data.build.load_inference_source(source=None, batch=1, vid_stride=1, buffer=False)
рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди рдХреЗ рд▓рд┐рдП рдПрдХ рдЕрдиреБрдорд╛рди рд╕реНрд░реЛрдд рд▓реЛрдб рдХрд░рддрд╛ рд╣реИ рдФрд░ рдЖрд╡рд╢реНрдпрдХ рдкрд░рд┐рд╡рд░реНрддрди рд▓рд╛рдЧреВ рдХрд░рддрд╛ рд╣реИред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
source |
(str, Path, Tensor, Image, ndarray)
|
рдЕрдиреБрдорд╛рди рдХреЗ рд▓рд┐рдП рдЗрдирдкреБрдЯ рд╕реНрд░реЛрддред |
None
|
batch |
int
|
рдбреЗрдЯрд╛рд▓реЛрдбрд░ рдХреЗ рд▓рд┐рдП рдмреИрдЪ рдЖрдХрд╛рд░ред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 1 рд╣реИред |
1
|
vid_stride |
int
|
рд╡реАрдбрд┐рдпреЛ рд╕реНрд░реЛрддреЛрдВ рдХреЗ рд▓рд┐рдП рдлреНрд░реЗрдо рдЕрдВрддрд░рд╛рд▓ред рдбрд┐рдлрд╝реЙрд▓реНрдЯ 1 рд╣реИред |
1
|
buffer |
bool
|
рдирд┐рд░реНрдзрд╛рд░рд┐рдд рдХрд┐рдпрд╛ рдЧрдпрд╛ рдХрд┐ рд╕реНрдЯреНрд░реАрдо рдлрд╝реНрд░реЗрдо рдмрдлрд░ рдХрд┐рдП рдЬрд╛рдПрдВрдЧреЗ рдпрд╛ рдирд╣реАрдВред рдбрд┐рдлрд╝реЙрд▓реНрдЯ рдЧрд╝рд▓рдд рд╣реИ. |
False
|
рджреЗрддрд╛:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо |
---|---|---|
dataset |
Dataset
|
рдирд┐рд░реНрджрд┐рд╖реНрдЯ рдЗрдирдкреБрдЯ рд╕реНрд░реЛрдд рдХреЗ рд▓рд┐рдП рдПрдХ рдбреЗрдЯрд╛рд╕реЗрдЯ рдСрдмреНрдЬреЗрдХреНрдЯред |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/build.py
2023-11-12 рдмрдирд╛рдпрд╛ рдЧрдпрд╛, рдЕрдкрдбреЗрдЯ рдХрд┐рдпрд╛ рдЧрдпрд╛ 2024-05-08
рд▓реЗрдЦрдХ: рдмреБрд░рд╣рд╛рди-рдХреНрдпреВ (1), рд▓рд╛рдлрд┐рдВрдЧ-рдХреНрдпреВ (1), рдЧреНрд▓реЗрди-рдЬреЛрдЪрд░ (3)