This repository is containing research code (read: not intended to be understood by anyone else than the creators).
Still, the functionality you might be looking for is also made available in a more curated repository. Check out CAREamics and MicroSplit reproducibility repo
Here, our goal is to disentangle the images by taking as input the sum of the two images. For example, given image A & B, input to our model is A+B and the output is A,B