Could you please release training codes of the whole pipeline?

[zhijiew]

Hi! Thank you for your great work which inspired me a lot.

I'm trying to follow your work, while I found that there are only codes of the online optimization training in this repository (which is the final step of the whole workflow). Can you release the training codes of other two steps (including initial flow decomposition network training and the layer reconstruction networks training)?

[filipppp]

A requirements.txt would be much appreciated.

[FionaFN]

Could you also please release the code and checkpoints for waterdrop removal? Thanks!