完整代码仓库: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加载数据
