mask_rcnn_infer_dl: Mask RCNN infer dataloader

View source: R/icevision_mask_rcnn.R

mask_rcnn_infer_dlR Documentation

Mask RCNN infer dataloader

Description

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

Usage

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


fastai documentation built on March 21, 2022, 9:07 a.m.