sketch-codeなる、ワイヤーフレームからhtmlを自動生成するAIツールを試してみた。

コードはこちら

#セットアップ
$ git clone https://github.com/ashnkumar/sketch-code.git
$ cd sketch-code/
$ pip install --upgrade pip #pip ver10が必要らしいです。
$ pip install -r requirements.txt

#モデルデータとかを取得
$ cd scripts/
$ sh get_data.sh
$ sh get_pretrained_model.sh
$ cd ../src
# ../examples/nash_ex02.pngを適切なパスに変更
$ python convert_single_image.py --png_path ../examples/nash_ex02.png --output_folder ./generated_html --model_json_file ../bin/model_json.json --model_weights_file ../bin/weights.h5 

こんな手書きのワイヤーが

こんな感じに出力されました。

４カラムじゃないよ！と言いたい気持ちを抑えつつ、
エリア定義ぐらいはいけそうです。

今回はpre-trained なモデルをダウンロードして使ってみましたが、
自前で学習させればもっといいのができるんでしょうね。

元記事はこちら
https://blog.insightdatascience.com/automated-front-end-development-using-deep-learning-3169dd086e82