基于pytorch的人工智能分类垃圾桶

2024-04-21 21:32:19

Hello，大家好，作者终于考完研了，现在开始更新自己以前的科研项目来供大家一起学习参考，开源共享，作者github网址：https://github.com/czzq1999，欢迎加油一起学习，一起开源中国。今天共享的开源项目是，利用python中的pytorch深度学习框架来实现全自动垃圾分类。

Introduction

本项目是基于YOLOv3的5G嵌入式智能分类垃圾桶系统。该垃圾桶图像处理具有两个接口选项，可以通过连接本地USB摄像头获取图像，也可以通过5G网络获取网络摄像头的图像。需要预先获取5G摄像头的IP地址以及端口。

Requirements

关于python依赖库安装使用如下命令：

pip3 install -r requirements.txt

如需移植到自己的电脑，需要修改cap.py文件里面的相关路径名，以及USB串口的路径。

需预先将单片机代码烧录进51单片机中，如何通过串口连接服务器。

如图为整个系统的框架图，需要用到两个数码舵机和一个工业摄像头，以及一个5V的稳压模块，和一个电源模块，以及一个单片机最小模块。

Training

Start Training：python3 train.py to begin training after downloading the dataset.
Resume Training：python3 train.py --resume to resume training from weights/xxx.pt

如图为训练loss损失曲线。

Dataset

我们收集了一千多张常见的垃圾种类数据集，包括金属，塑料，纸质，玻璃，通过图像增广，获得了几万张图片。

每类图片有三百张左右。

本项目的代码已经上传至Github上：https://github.com/czzq1999/Garbage-detection-and-classification

基于pytorch的人工智能分类垃圾桶相关推荐

基于Pytorch的cifar10分类网络模型
Pytorch作为新兴的深度学习框架,目前的使用率正在逐步上升.相比TensorFlow,Pytorch的上手难度更低,同时Pytorch支持对图的动态定义,并且能够方便的将网络中的tensor格式数 ...
基于Pytorch实现的流式与非流式语音识别
前言 MASR是一款基于Pytorch实现的语音识别框架,MASR致力于简单,实用的语音识别项目.可部署在服务器,Nvidia Jetson设备,未来还计划支持Android等移动设备. 源码地址:h ...
基于pytorch搭建神经网络的花朵种类识别（深度学习）
基于pytorch搭建神经网络的花朵种类识别(深度学习) 文章目录基于pytorch搭建神经网络的花朵种类识别(深度学习) 一．知识点 1.特征提取.神经元逐层判断 2.中间层(隐藏层) 3.学习权 ...
Deep Learning：基于pytorch搭建神经网络的花朵种类识别项目（内涵完整文件和代码）—超详细完整实战教程
基于pytorch的深度学习花朵种类识别项目完整教程(内涵完整文件和代码) 相关链接:: 超详细--CNN卷积神经网络教程(零基础到实战) 大白话pytorch基本知识点及语法+项目实战文章目录基 ...
【项目实战课】基于Pytorch的EfficientNet血红细胞分类竞赛实战
欢迎大家来到我们的项目实战课,本期内容是<基于Pytorch的EfficientNet血红细胞分类竞赛实战>.所谓项目课,就是以简单的原理回顾+详细的项目实战的模式,针对具体的某一个主题, ...
基于pytorch后量化（mnist分类）---浮点训练vs多bit后量化vs多bit量化感知训练效果对比
基于pytorch后量化(mnist分类)-浮点训练vs多bit后量化vs多bit量化感知训练效果对比代码下载地址:下载地址试了 bit 数为 1-8 的准确率,得到下面这张折线图: 发现,当 b ...
艾伦人工智能研究院开源AllenNLP，基于PyTorch轻松构建NLP模型
艾伦人工智能研究院(AI2)开源AllenNLP,它是一个基于PyTorch的NLP研究库,利用深度学习来进行自然语言理解,通过处理低层次的细节.提供高质量的参考实现,能轻松快速地帮助研究员构建新的语 ...
基于PyTorch搭建CNN实现视频动作分类任务代码详解
数据及具体讲解来源: 基于PyTorch搭建CNN实现视频动作分类任务 import torch import torch.nn as nn import torchvision.transforms ...
Python基于PyTorch实现BP神经网络ANN分类模型项目实战
说明:这是一个机器学习实战项目(附带数据+代码+文档+视频讲解),如需数据+代码+文档+视频讲解可以直接到文章最后获取. 1.项目背景在人工神经网络的发展历史上,感知机(Multilayer Per ...

最新文章

热门文章