ProsePainter是一款基于机器学习的有趣开源工具,可以通过文字描述来生成图像。遵守GPL3.0开源协议。

程序包含一个python服务器用来计算,还有一个使用nodejs服务器的网页端界面 .用户只需要描述想画的东西和在那个区域,ProsePainter就会根据机器学习的实时计算生成图片。

安装运行:

安装Python 服务器

建议使用e python3.7 以及 torch 1.7.1+cu110 以上版本(这里下载 here).

建立 conda 环境,取名 prosepaint.

conda env create -f server/env-server.yml

运行 Python 服务器

运行命令启动 uvicorn 服务器.

export PYTHONPATH=.; python server/server_deploy.py

安装NodeJS 服务器

cd clienvbt
npm install

运行NodeJS 服务器

cd client
npm run dev

浏览器打开 http://localhost:8003/ 即可使用。

源码:https://github.com/Morphogens/ProsePainter