The work carried out a study on modifying the architecture of the TransMorph neural network by integrating an input data preprocessing unit. The goal was to achieve better similarity scores between studies in the dataset and the  reference atlas. The data was assessed based on the structure similarity metric. The results suggest that the use of a Sobel filter can lead to improvement.