最近的上海人
没有被房价逼退
没有被快节奏工作压力打倒,
却在垃圾分类上栽了跟头...
(图片来自网络)
“垃圾分类”这个毫无网络气息的词汇,像病毒一样传播着。“你是什么垃圾”,每天都要接受阿姨的灵魂拷问!
这样的日子还要忍多久?别慌,程序员一段代码,什么分类都能搞定!
1. 工具包导入
这⾥我们选⽤比较友好的 Keras 工具包帮我们搭建模型。
2. 读取图像数据
3. 建立视觉模型
搭建CNN 模型,这⾥我们⽤了三层卷机模型。感兴趣的⼩伙伴可以试试看 VGG16
4. 训练视觉模型
燃烧小宇宙的训练时间来了
5. 验证模型结果
最终我们来看看这个结果,读取测试用的图像数据,进行预测,再跟实际进⾏对比。
6. 最后,将它放在后台,⽤小程序或是 App 的⽅式,只要拿出⼿机扫⼀扫就能知道“它是个什么垃圾”了。