How to implement Mosaic or MixUp Augmentation to the Supervised and Strong Pipeline ?

[oracl4]

I try to implement the mosaic and mixup augmentation in your code, but I confused because the dataset is wrapped around the SemiDataset wrapper and the mosaic or mixup augmentation need the dataset to wrap on the MultiImageMixDataset type. Do you have any idea how to implement this kind of work ?

[MendelXu]

Is it possible to implement it like this?https://github.com/open-mmlab/mmclassification/blob/188aa6ed5dae184024818a6c3a72e5b0c733888d/mmcls/models/classifiers/image.py#L103