本项目为智能垃圾桶的代码仓库，包含：
- 垃圾分类模型训练与推理 (基于YOLOv11n)
- 上位机交互系统 (基于Pyside6和Serial通信)
./
│ 📄 amcap v3.0.9.exe # 🎥 摄像头调试工具
│ 📄 requirements.txt # 📦 Python依赖库
│
├─🧠 garbage_train/ # 🤖 模型训练和转换模块
│ │ 📄 convert.py # 🔄 模型格式转换（pt -> onnx）
│ │ 📄 dataset.yaml # 📊 数据集配置
│ │ 📄 data_copy.py # ⎘ 数据集复制工具
│ │ 📄 data_extension.py # ➕ 数据集增强脚本
│ │ 📄 garbage_type.txt # 🏷️ 垃圾类别标签
│ │ 📄 infer_label.py # 🏷️ 标签推理工具
│ │ 📄 modify_file_name.py # ✏️ 文件名修改工具
│ │ 📄 random_pick.py # 🎲 划分训练集和验证集
│ │ 📄 README.md # 📝 打标签和模型训练全过程说明
│ │ 📄 relabel_tool.py # 🔖 重新标注qt工具
│ │ 📄 train.py # 🏋️ 训练脚本
│ │ 📄 yolo11n.pt # ⚙️ YOLOv11预训练权重
│ │ 📄 yolov8n.pt # ⚙️ YOLOv8预训练权重
│ │
│ └─🗃️ data/
│ └─dataset_refine/
│ ├─📂 images/
│ ├─📂 train/ # 🖼️ 训练图片
│ └─📂 val/ # 🖼️ 验证图片
│ └─📂 labels/
│ ├─📂 train/ # 📝 训练标签
│ └─📂 val/ # 📝 验证标签
│
└─💻 wobudaoa/ # 🖥️ 上位机系统
│ 📄 coordinate_picker.py # 🎯 坐标提取工具
│ 📄 visualize_layout.py # 🎨 布局可视化
│ 📄 debug_headless.py # 🐛 无界面调试模式
│ 📄 model_test.py # 🧪 模型测试脚本
│ 📄 logic_thread.py # 🤔 视觉处理线程（测试）
│ 📄 trans_test.py # 🧪 通信协议（测试）
│ 📄 qt_test.py # 🧪 主界面逻辑（测试）
│ 📄 yolo_thread.py # 👁️ 视觉处理线程
│ 📄 yolo_thread_chusai.py # 🏆 视觉处理线程（初赛专用）
│ 📄 trans.py # 🔄 通信协议
│ 📄 qt_backend.py # 🖌️ 主界面逻辑
│
├─📂 backups/
│ ├─📂 84决赛前测试单垃圾不改动/
│ │ 📄 qt_backend.py
│ │ 📄 trans.py
│ │ 📄 yolo_thread_chusai.py
│ │
│ └─📂 84决赛前调试多垃圾稳定能运行决策/
│ 📄 logic_thread.py
│ 📄 qt_test.py
│ 📄 trans_test.py
│
├─📂 camera/
│ 📄 combine.py # 🎥 拍照+模型推理+可视化
│
├─📂 img/
│ 📄 cover.jpg # 🖼️ 封面图
│ 📄 cqu.png # 🏫 校徽LOGO
│ 📄 garbageicon.png # 🗑️ 垃圾图标
│ 📄 logo.png # 🔷 项目LOGO
│
├─📂 model/
│ 📄 test82_refine_ex.onnx # 🧠 优化模型v82
│ 📄 test83_refine_ex.onnx # 🧠 优化模型v83
│
└─📂 video/
📄 garbag.mp4 # 📹 演示视频
# 安装依赖
pip install -r requirements.txt# 运行上位机系统
python wobudaoa/qt_backend.py本项目采用 MIT License
重庆大学工创赛团队 我不到啊 @2025