View source: R/icevision_dataloaders.R
mask_rcnn_valid_dl | R Documentation |
A 'DataLoader' with a custom 'collate_fn' that batches items as required for training the model.
mask_rcnn_valid_dl(dataset, batch_tfms = NULL, ...)
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. |
None
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.