aeneonlxc/baidu_easydl

2022人工智能挑战赛项目代码

2022人工智能挑战赛项目代码

说明：本项目的数据集未开源，是我本科的Mentor自己标注的数据集，原数据集总量有10450张，数据集的标签有cargo（包裹），tray（货架）以及forklift（叉车），因参赛需要上传数据集的原因，经过Mentor授权，我可使用2000张数据集作为训练学习使用。本项目代码仅供学习交流使用。

原始数据集不考虑开源

有问题请联系我的邮箱：[email protected]

更新公告!!

---

• 【2022/09/28】 项目获得西南赛区区域三等奖，虽然拉，但有证书。
• 【2022/08/15】 支持视频检测以及图像检测。

项目简介

---

基于仓储物流识别的人工智能挑战项目，本仓库仅仅有训练代码及相关文档，不包含项目本身的所有程序。

项目结构

---

项目分为图像分类以及图像检测两个板块

project
│  baidu_api.py 									#封装好的百度API实例
│  getdata.py									    #数据清洗（伪）
│  img_deticion.py									#视频流的图像分类
│  img_classfication.py								#视频流的图像检测
│  README.md									    #说明文档
│  基于百度EasyDL的仓储物流识别技术文档V1.0-李星辰.docx  #技术文档

项目环境

---

​ 注意：本项目所用到的深度学习环境为实验室主机的深度学习环境，我在主机上预装了pytorch以及tensorflow，均能够使用GPU训练深度学习模型，务必慎用，如果环境混乱或者错误请安装本文档的环境版本进行重装，目前未备份镜像。

Software	Version	Hardware	Version
Ubuntu	18.04	GeForce RTX3080	七彩虹
Windows	10
Python	3.7.10
百度EasyDL平台	2022.08.12
cuda	11.3.0_465_19_01
cudnn	8.2.0.53-1
pytorch	1.12.1
torchvision	0.13.1
torchaudio	0.12.1
tensorflow-gpu	2.7.0