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

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

项目地址

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

使用例子

example.png

如何使用

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

10 回复

厉害啦~支持

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

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

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

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

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

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

回到顶部