How did you make it work for more than just half half division?

[alexblattner]

Hi @opparco , I have been trying to make two shot work for any division, but when I don't do the same format as the one in the colab, the image is broken.

I know you didn't make that colab, but I was wondering how you managed to solve that issue

edit: you must make sure all the picture is occupied with masks