在 Node.js 平台上用 JavaScript 尝试玩深度学习~ 基于TensorFlow.js。
项目地址
https://github.com/DavidCai1993/style-transfer.js
使用例子
如何使用
详情见于项目介绍。目前仅支持 GPU 。
回复 (11)
Y
厉害啦~支持
A
666666
D
@i5ting 对,例子都很有启发~
S
请问你这个模型是怎么生成,python生成的 Frozen Model 转化成tensorflowjs的模型 有.pb文件和.json文件,貌似没法直接load....
A
不错,学习了
I
在google io大会tfjs相当惊艳,很好的尝试
D
@yorkie 蛤蛤,谢谢 Yorkie 老师支持~
N
有意思
D
@shuaishenk 用了这个工具:https://github.com/tensorflow/tfjs-converter
S
引用 DavidCai1993@shuaishenk 用了这个工具:https://github.com/tensorflow/tfjs converter
@DavidCai1993 用这个转过的模型,js输出的结果跟python不一样,输入的预处理后的数据是一致的。
S
我找到了这个库 https://github.com/ml5js ,用它训练了模型,并导出了自定义的,基于mobilenet模型的。 用的时候直接用tsjs就可以了,先加载mobilenet,提取出微调层以上的输出,作为自定义模型的输入,结果没问题了
参与回复
登录后即可参与回复。登录