/ 数据分析

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

项目地址
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

-----2018-11-11---4.35.11
-----2018-11-11---4.35.47
然后需要有pip模块安装,就像 node 的 npm;

安装依赖

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

pip install -r requirements.txt

-----2018-11-11---4.36.02
-----2018-11-11---4.36.22

之后根据项目说明,依次进行

下载数据与训练权重

(数据量大,下载比较费时,找一个网络好的地方)

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

原型

-----2018-11-11---6.50.29
绘制结果
-----2018-11-11---6.50.05

我是直接用的项目里面的图
examples目录下
对应生成
generated_html目录下
-----2018-11-11---6.50.19