CNode

在 Node.js 里尝鲜玩深度学习之画风变换

精华
分享
DDavidCai1993发布于8 年前最后回复8 年前11 回复12121 浏览0 收藏

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

项目地址

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

使用例子

example.png

如何使用

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

查看回复

回复 (11)

Y
yorkie#1·8 年前

厉害啦~支持

A
aojiaotage#2·8 年前

666666

D
DavidCai1993#3·8 年前

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

S
shuaishenk#4·8 年前

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

A
a69694510#5·8 年前

不错,学习了

I
i5ting#6·8 年前

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

D
DavidCai1993#7·8 年前

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

N
NGFIX#8·8 年前

有意思

D
DavidCai1993#9·8 年前

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

S
shuaishenk#10·8 年前
引用 DavidCai1993@shuaishenk 用了这个工具:https://github.com/tensorflow/tfjs converter

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

S
shuaishenk#11·8 年前

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

参与回复
登录后即可参与回复。登录