【人工智能项目】基于paddlepaddle平台的安全帽检测比赛

Frank牛蛙

已于 2024-07-06 22:42:15 修改

阅读量2.3k

点赞数 55

分类专栏：人工智能文章标签： python 目标检测 paddlepaddle

于 2024-05-24 20:31:53 首次发布

本文链接：https://blog.csdn.net/weixin_64259675/article/details/139180191

版权

一、问题描述

共包含3类：0:'head', 1:'helmet', 2:'person' 提交格式要求，提交名为pred_result.txt的文件:每一行代表一个目标，每一行内容分别表示图像名置信度 xmin ymin xmax ymax 类别

paddlepaddle平台

算力尽量选择高一些的，跑模型花的时间少一些

主要应用了paddleX以及 PaddleDetection-2.5.0，paddleX主要用在了数据分割，数据增强等数据预处理的过程中；PaddleDetection-2.5.0中选择了一个模型，没能完全运行完，

模型文件（包含完整路径）为

/home/aistudio/PaddleDetection-2.5.0/configs/yolox/yolox_cdn_tiny_300e_coco.yml，我跑了40epoch，共300epoch，跑完大概需要六七个小时，时间充裕建议跑完

我的数据集存储路径：

导库

!pip install paddlex
!pip show paddlex

分割数据集

!paddlex --split_dataset --format VOC --dataset_dir /home/aistudio/helmet-dataset/train --val_value 0.15

此时可能归遇到的问题：

这时需要运行

!pip install --upgrade pip
!pip install --user --upgrade pyarrow==11.0.0

之后就可以成功运行上面分割数据集的代码啦

运行结果截图

建议代码生成上述文件&#