μ°Έμ‘° ultralytics/data/build.py
μ°Έκ³
μ΄ νμΌμ https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/data/build .pyμμ νμΈν μ μμ΅λλ€. λ¬Έμ λ₯Ό λ°κ²¬νλ©΄ ν 리νμ€νΈ (π οΈ) λ₯Ό ν΅ν΄ λ¬Έμ λ₯Ό ν΄κ²°νλλ‘ λμμ£ΌμΈμ. κ°μ¬ν©λλ€ π!
ultralytics.data.build.InfiniteDataLoader
λ² μ΄μ€: DataLoader
μμ μλ₯Ό μ¬μ¬μ©νλ λ°μ΄ν° λ‘λ.
λ°λλΌ λ°μ΄ν°λ‘λμ λμΌν ꡬ문μ μ¬μ©ν©λλ€.
μ μμ€ μ½λ ultralytics/data/build.py
__init__(*args, **kwargs)
μμ μλ₯Ό 무νν μ¬νμ©νλ λ°μ΄ν°λ‘λλ λ°μ΄ν°λ‘λμμ μμν©λλ€.
μ μμ€ μ½λ ultralytics/data/build.py
__iter__()
__len__()
reset()
λ°λ³΅κΈ°λ₯Ό μ¬μ€μ ν©λλ€.
μ΄ κΈ°λ₯μ νμ΅ μ€μ λ°μ΄ν° μΈνΈμ μ€μ μ μμ νλ €λ κ²½μ°μ μ μ©ν©λλ€.
ultralytics.data.build._RepeatSampler
μμν λ°λ³΅λλ μνλ¬.
맀κ°λ³μ:
μ΄λ¦ | μ ν | μ€λͺ | κΈ°λ³Έκ° |
---|---|---|---|
sampler |
sampler
|
μνλ¬λ₯Ό λ°λ³΅ν©λλ€. |
νμ |
μ μμ€ μ½λ ultralytics/data/build.py
__init__(sampler)
ultralytics.data.build.seed_worker(worker_id)
λ°μ΄ν°λ‘λ μ컀 μλ μ€μ https://pytorch.org/docs/stable/notes/randomness.html#dataloader.
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)
νλ ¨ λλ μ ν¨μ± κ²μ¬ μ§ν©μ μν 무ν λ°μ΄ν° λ‘λ λλ λ°μ΄ν° λ‘λλ₯Ό λ°νν©λλ€.
μ μμ€ μ½λ 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μ λλ€. |
False
|
λ°νν©λλ€:
μ΄λ¦ | μ ν | μ€λͺ |
---|---|---|
dataset |
Dataset
|
μ§μ λ μ λ ₯ μμ€μ λν λ°μ΄ν° μΈνΈ κ°μ²΄μ λλ€. |
μ μμ€ μ½λ ultralytics/data/build.py
μμ± 2023-11-12, μ λ°μ΄νΈ 2024-05-08
μμ±μ: Burhan-Q (1), μλ-ν (1), κΈλ -μ‘°μ² (3)