Tài liệu tham khảo cho ultralytics/data/build.py
Ghi
Tệp này có sẵn tại https://github.com/ultralytics/ultralytics/blob/main/ultralytics/dữ liệu/build.py. Nếu bạn phát hiện ra một vấn đề, vui lòng giúp khắc phục nó bằng cách đóng góp Yêu cầu 🛠️ kéo. Cảm ơn bạn 🙏 !
ultralytics.data.build.InfiniteDataLoader
Căn cứ: DataLoader
Bộ nạp dữ liệu tái sử dụng công nhân.
Sử dụng cú pháp tương tự như vani DataLoader.
Mã nguồn trong ultralytics/data/build.py
__init__(*args, **kwargs)
Dataloader tái chế vô hạn công nhân, kế thừa từ DataLoader.
Mã nguồn trong ultralytics/data/build.py
__iter__()
__len__()
reset()
Đặt lại bộ lặp.
Điều này rất hữu ích khi chúng tôi muốn sửa đổi cài đặt của tập dữ liệu trong khi đào tạo.
ultralytics.data.build._RepeatSampler
Bộ lấy mẫu lặp đi lặp lại mãi mãi.
Thông số:
Tên | Kiểu | Sự miêu tả | Mặc định |
---|---|---|---|
sampler |
sampler
|
Bộ lấy mẫu để lặp lại. |
bắt buộc |
Mã nguồn trong ultralytics/data/build.py
__init__(sampler)
ultralytics.data.build.seed_worker(worker_id)
Đặt hạt giống thợ dataloader 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)
Xây dựng YOLO Tập dữ liệu.
Mã nguồn trong ultralytics/data/build.py
ultralytics.data.build.build_grounding(cfg, img_path, json_file, batch, mode='train', rect=False, stride=32)
Xây dựng YOLO Tập dữ liệu.
Mã nguồn trong ultralytics/data/build.py
ultralytics.data.build.build_dataloader(dataset, batch, workers, shuffle=True, rank=-1)
Trả về InfiniteDataLoader hoặc DataLoader cho tập hợp đào tạo hoặc xác thực.
Mã nguồn trong ultralytics/data/build.py
ultralytics.data.build.check_source(source)
Kiểm tra loại nguồn và trả về các giá trị cờ tương ứng.
Mã nguồn trong ultralytics/data/build.py
ultralytics.data.build.load_inference_source(source=None, batch=1, vid_stride=1, buffer=False)
Tải nguồn suy luận để phát hiện đối tượng và áp dụng các phép biến đổi cần thiết.
Thông số:
Tên | Kiểu | Sự miêu tả | Mặc định |
---|---|---|---|
source |
(str, Path, Tensor, Image, ndarray)
|
Nguồn đầu vào để suy luận. |
None
|
batch |
int
|
Kích thước hàng loạt cho bộ tải dữ liệu. Mặc định là 1. |
1
|
vid_stride |
int
|
Khoảng thời gian khung hình cho các nguồn video. Mặc định là 1. |
1
|
buffer |
bool
|
Xác định xem khung sự kiện trực tiếp có được lưu vào bộ đệm hay không. Mặc định là False. |
False
|
Trở lại:
Tên | Kiểu | Sự miêu tả |
---|---|---|
dataset |
Dataset
|
Một đối tượng tập dữ liệu cho nguồn đầu vào được chỉ định. |