Image Registration with python

I have an image dataset that consist of different exposed(3 exposed images: low-medium-high) images and i have a registration network publicly available. I want this network to be trained and tested with this dataset. Registration task will be trained seperately on low-medium and high-medium exposed image pairs