DetectionEngine.loader

DetectionEngine.loader(
    model,
    split,
    *,
    anchors=None,
    shuffle=False,
    augmented=False,
)

A torch DataLoader over one split, wrapping dataset.

Parameters

Name	Type	Description	Default
model	DetectionModel	Which model’s settings to read.	required
split	str	`"train"`, `"validation"` or `"test"`.	required
anchors		As for `dataset`.	`None`
shuffle	bool	Shuffle between epochs. Training only.	`False`
augmented	bool	Apply the model’s augmentation. Training only.	`False`

Name	Type	Description
		A `torch.utils.data.DataLoader` yielding images and encoded targets.