Testing on the brain, a less homogenenous area

[BreziTasbi]

Since the current model cannot do much more than finding the correct value to color the spinal coord with, i am going to try to find a Healthy brain dataset in order to train a new model on it and see if the method can be more relevant on a it.
Thus i have to gather:
-A healthy brain dataset with a couple hundreds of subjects.
-A brain dataset with similar acquisition parameters (contrast, resolution) and visible lesions.

[BreziTasbi]

As the brain cannot be reduced to a 32x256x256 volume, I tried to reduce lower the latent space size by a factor 2 on each dimension shape in order to allow the management of 256x256x256 volumes. Event if this actually is the recommended latent space size in original paper, I have very blurry results that cannot be considered descent counterfactual.

I should stick to a larger latent space size and try to center on the brain better in order to try 128x128x128. Nevertheless, it means that the entire brain cannot be processed at once.