完整代码仓库:https://github.com/252547545/waste_sorting.git
建议
拓展
垃圾分类网站
utils只img2x方法
app.js
this.CLASSES
app.js
utils.js
App.js中引入tfjs,并加载模型
http-server启动一个静态服务去加载模型
npm i @temsorflow/tfjs -S
app.js
index.js
app.js
安装打包工具parcel
操作步骤
tf.util.shuffle(data)#洗牌
图片混着学习更高效
分批训练
img2x
运行代码
训练模型、保存模型
epochs:训练多少轮
操作步骤
截断模型、替换为自己的网络
截断模型
从conv_pw_13_reu这层开始截断
pipcook
操作步骤
迁移训练
data.js
slice少测试一点数据
index.js加载数据