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