数据集包含六种类型的垃圾： cardboard、glass、metal、paper、plastic和其他废品，数据量很少，因此仅用于学习。
数据集的标准准备工作，包括将数据集分为训练集和测试集并创建标记文件。 代码utils.py
importosimportshutilimportjsonpath=' e ://dataset//garbage _ class ification ' #此路径是上图中的六种目录， 根据您数据集的路径，classes=[ garbageforgarbageinos.list dir (path ) ]ifOS.path.exists(OS.path.join ) OS.getcwd )==false : OS.makedirs (OS.path.join ) OS.getcwd (，' train ' ) ) ifos.path.exists (OS.patis ts ) ) )。 ' val ' )==false : OS.makedirs (OS.path.join (OS.getcwd )，' val ' ) ) f=open (garbage _ train.j soon