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

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

项目地址

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

yorkie 1楼•6 年前

厉害啦～支持

aojiaotage 2楼•6 年前

666666

i5ting 3楼•6 年前

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

DavidCai1993 4楼•6 年前作者

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

DavidCai1993 5楼•6 年前作者

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

NGFIX 6楼•6 年前

有意思

shuaishenk 7楼•6 年前

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

DavidCai1993 8楼•6 年前作者

shuaishenk 9楼•6 年前

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

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

a69694510 11楼•6 年前

不错，学习了