最近tensorflowjs更新了,然后这篇的方法似乎有点行不通,如果失效请参考新的博客:Tensorflow.js运行Python下训练的CNN模型
一、引言
这两天的项目需要用到Tensorflow.js来实现一个AI,尽管说Tensorflow.js本身是有训练模型的功能的,不过考虑到javascript这个东西加载资源要考虑跨域问题等种种因素。。最终还是决定使用python的tensorflow来训练模型,然后利用js端来使用模型进行运算,那么关键问题就是:js如何加载python下训练的模型
【webAI】Tensorflow.js加载预训练的model
这位博主的博客给了我很大的帮助,前两步按照他的教程来做都是没有什么问题的,不过其实还是有一些潜在的坑或者对于我这种前端小白不太友好的地方,这里我把我的整个过程都来叙述一遍吧。
注:首先在命令行中执行
pip install tensorflowjs
安装模型转换的部分,否则转换可能会报错
二、python部分
这里我用了一个更加简单的例子,and.py,让神经网络来学习异或运算(忽略这个"and"emmm),这里我直接把python代码贴出来:
#coding=utf-8#
import tensorflow as tf
import numpy as np
x_data=[[0.0,0.0],[0.0,1.0],[1.0,0.0],[1.0,1.0]] #训练数据
y_data=[[0.0],[1.0],[1.0],[0.0]] #标签
x_test=[[0.0,1.0],