Using all three color spaces

[chokevin8]

So I calculated my dataset statistics by using dataset_statistics.py for all three color spaces (RGB, HSV, HED), and then created a yaml file that looked exactly like this. When trying to use this, do I just fix this line (line 50) so that self.color_space is selected randomly out of the three color spaces? Then, when I put it inside my transforms function like below, the transforms will select one color space, augment & normalize and then re-normalize with Reinhard method? I'd appreciate some help with this, thanks!

#randstain_all.yaml is my yaml folder val_transforms = transforms.Compose([RandStainNA( yaml_file="randstainna_all.yaml", std_hyper=-0.3, probability=0.8, distribution="normal", is_train=True )])