回收网
在大量生产、大量消费的时代,垃圾处理已经成为一个重要的国家课题。 随着这一趋势,垃圾收集和再利用的社会和经济重要性正在增加。 一种替代方法是,一旦用户排出垃圾,无论材料如何,机器都可以自动分类。
使用两种方法创建有效的垃圾分类模型,仅使用少量带注释的垃圾图像(2527)。
1)迁移学习:使用ImageNet预训练模型2) 使用注意力模块进行有效的特征学习
为了证明所提出的方法是有效的,进行了大量的消融研究,并且比最先进的注意力模块更有效。
骨干网络:ResNet
注意模块:RecycleNet
要求
使用 pip 安装所有 python 依赖项:
$ git clone https://github.com/sangminwoo/RecycleNet.git
$ cd RecycleNet
$ pip install -r requirements.txt
PyTorc
评论0
最新资源