How to apply Group Normalization to PyTorch FasterRCNN training?

vision
1

I am following this implementation of RCNN (I will use FasterRCNN instead of MaskRCNN) https://pytorch.org/tutorials/intermediate/torchvision_tutorial.html.

I want to use Group Normalization for the input layer in my DataSet class. I don’t want to use custom pipeline and create my DataSet and DataLoader since PyTorch training pipeline is very convenient to use.

Is it possible to use GroupNormalization with standard Dataloader and Dataset from torch.utils.data?

There is one (I think) available DataLoader: -> https://pytorch.org/docs/stable/data.html#torch.utils.data.DataLoader and couple DataSets: -> Dataset: https://pytorch.org/docs/stable/data.html#torch.utils.data.Dataset -> IterableDataset: https://pytorch.org/docs/stable/data.html#torch.utils.data.IterableDataset …

Thanks!