Video foreground extractor by end-to-end training a UNet train python model.py test python model_test.py