This is a demo frontend of "Learning to Paint", a Reinforcement Learning based painting agent. The demo applied pre-trained models from ICCV2019-LearningToPaint, including both actor and renderer.
Use the package manager conda to install dependencies.
conda install -r requirements.txt -n learningtopaint
conda activate learningtopaintInstall ffmpeg, for Mac users:
brew install ffmpegpython setup.py install
export FLASK_APP=app
export FLASK_DEBUG=true
flask runGo to http://127.0.0.1:5000 (allow camera to be used in browser)
Run tests:
pytest -v
Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.
Please make sure to update tests as appropriate.