A `DataLoader` with a custom `collate_fn` that batches items as required for inferring the model.

mask_rcnn_infer_dl(dataset, batch_tfms = NULL, ...)

Arguments

dataset

Possibly a `Dataset` object, but more generally, any `Sequence` that returns records.

batch_tfms

Transforms to be applied at the batch level. **dataloader_kwargs: Keyword arguments that will be internally passed to a Pytorch `DataLoader`. The parameter `collate_fn` is already defined internally and cannot be passed here.

...

additional arguments

Value

None