这是一个使用PyTorch实现的垃圾分类系统,支持CNN和ResNet两种模型,并提供了完整的PyQt5图形界面。

功能特性

        多模型支持: 支持CNN和ResNet两种深度学习模型

        图片预测: 支持单张图片的垃圾分类预测

        可视化: 提供预测概率分布图表

        友好界面: 基于PyQt5的现代化图形界面

        模型管理: 支持加载自定义训练的模型

垃圾分类类别  系统支持以下6种垃圾分类:

        Cardboard (纸板)

        Glass (玻璃)

        Metal (金属)

        Paper (纸张)

        Plastic (塑料)

        Trash (垃圾)

更多推荐