精华 在 Node.js 里尝鲜玩深度学习之画风变换
发布于 2 个月前 作者 DavidCai1993 2822 次浏览 来自 分享

在 Node.js 平台上用 JavaScript 尝试玩深度学习~ 基于TensorFlow.js。

项目地址

https://github.com/DavidCai1993/style-transfer.js

使用例子

example.png

如何使用

详情见于项目介绍。目前仅支持 GPU 。

10 回复

厉害啦～支持

666666

1

在google io大会tfjs相当惊艳，很好的尝试

1

@yorkie 蛤蛤，谢谢 Yorkie 老师支持~

1

@i5ting 对，例子都很有启发~

有意思

请问你这个模型是怎么生成，python生成的 Frozen Model 转化成tensorflowjs的模型 有.pb文件和.json文件,貌似没法直接load…

@shuaishenk 用了这个工具：https://github.com/tensorflow/tfjs-converter

@DavidCai1993 用这个转过的模型，js输出的结果跟python不一样，输入的预处理后的数据是一致的。

我找到了这个库 https://github.com/ml5js ，用它训练了模型，并导出了自定义的,基于mobilenet模型的。 用的时候直接用tsjs就可以了，先加载mobilenet，提取出微调层以上的输出，作为自定义模型的输入，结果没问题了

回到顶部