Giter VIP home page Giter VIP logo

ai-painter-wx-miniprogram's Introduction

ai-painter-wx-miniprogram

tensorflow跑在微信小程序上,实现AI识别简笔画,AI自动画画等。
mini-code

quick_draw_classifier模型

说明

对用户画的简笔画进行分类。

数据集

使用quick_draw dataset.

模型训练

https://github.com/zhaocc1106/machine_learn/tree/master/NeuralNetworks-tensorflow/RNN/quick_draw
https://github.com/zhaocc1106/machine_learn/tree/master/NeuralNetworks-tensorflow/RNN/quick_draw/quick_draw_classify
训练完之后通过如下命令将模型转换为tensorflow.js模型。

tensorflowjs_converter --input_format keras model.h5 ./

转换后会生成model.json文件以及group bin文件,分别记录网络结构以及参数值。
实际应用模型时可能需要修改一些地方,比如修改model.json的input layer的batch size为1;修改dropout training参数为false。当然也可以在保存HDF5模型文件前做修改。

演示

ai-classifier

quick_draw_autopainter模型

说明

根据用户的起始笔画,来完成后续的简笔画.

数据集

使用quick_draw dataset.

模型训练

https://github.com/zhaocc1106/machine_learn/tree/master/NeuralNetworks-tensorflow/RNN/quick_draw
https://github.com/zhaocc1106/machine_learn/tree/master/NeuralNetworks-tensorflow/RNN/quick_draw/auto_draw
训练完之后通过如下命令将模型转换为tensorflow.js模型。

tensorflowjs_converter --input_format keras model.h5 ./

转换后会生成model.json文件以及group bin文件,分别记录网络结构以及参数值。
实际应用模型时可能需要修改一些地方,比如修改model.json的input layer的batch size为1;修改dropout training参数为false。当然也可以在保存HDF5模型文件前做修改。

遗留问题

对起始笔画要求比较高,可能是如下原因:

  1. 笔画采集和预处理不太满足模型输入
  2. 模型本身问题
  3. 训练数据不够随机性

后续再继续优化

演示

ai-painter

小程序中部署tensorflow模型

参考官方说明https://github.com/tensorflow/tfjs-wechat

ai-painter-wx-miniprogram's People

Contributors

zhaocc1106 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.