Simple Before / After U-Net model training for Flame
Its a real-life training script that has been tested in production several times already.
Dataset for command-line training can be exported from Autodesk Flame and training can be initiated with a command-line tool in background.
There is a GUI to apply trained model to a clips in Autodesk Flame.
Copy flameSimpleML folder to your flame's Python folder and refresh hooks.