The switch pipeline was not compatible with MultImageMixDataset: it only replaced the loading_pipeline of the outer (MultiImageMixDataset) dataset. With this modification, the loading pipeline is set to the identity function, and the inner (train_pipeline) is rewritten with the one provided to the hook.
Motivation
The switch pipeline was not compatible with MultImageMixDataset: it only replaced the
loading_pipelineof the outer (MultiImageMixDataset) dataset. With this modification, the loading pipeline is set to the identity function, and the inner (train_pipeline) is rewritten with the one provided to the hook.