利用机器学习把草图自动生成HTML

前朝楚水

发布于 2018-12-12 16:11:46

1.2K0

发布于 2018-12-12 16:11:46

文章被收录于专栏：互联网杂技

做了这个实验，看来只会切页面的前端，是可能被AI替代的。

项目地址 https://github.com/ashnkumar/sketch-code 可以先clone到本地

git clone https://github.com/ashnkumar/sketch-code

环境准备

要做此实验先按照说明把环境准备好，要python3.0以上；建议按照 pyenv 就像node 的 nvm 可以切换python的版本。

查看安装的python的版本

pyenv versions

设置

python local 3.5.2

然后需要有pip模块安装，就像 node 的 npm；

安装依赖

进入到项目的跟目录安装依赖

pip install -r requirements.txt之后根据项目说明，依次进行

下载数据与训练权重

（数据量大，下载比较费时，找一个网络好的地方）

cd scripts
sh get_data.sh
//可以手动自己下载这个数据，可能快点,下载之后放 data 目录
//http://sketch-code.s3.amazonaws.com/data/all_data.zip
sh get_pretrained_model.sh
//下载之后放 bin 目录
//http://sketch-code.s3.amazonaws.com/model_json_weights/model_json.json
//http://sketch-code.s3.amazonaws.com/model_json_weights/weights.h5

根据原型图，绘制

python convert_single_image.py --png_path ../examples/drawn_example2.png \
      --output_folder ./generated_html \
      --model_json_file ../bin/model_json.json \
      --model_weights_file ../bin/weights.h5

原型