目录
一、工具及环境
1.工具
labelimg
AI Studio
YOLO2COCO
PaddleUtils
paddleyolo
2.环境要求
openvino==2022.2.0
paddle2onnx==1.0.5
paddlepaddle==2.4.2
opencv-python==4.2.0.32
onnx==1.11.0
tensorflow==2.9.
3.工具介绍
labelimg是一个有图形界面的图像标注工具,用来给数据打标签。
AI Studio是基于百度深度学习平台飞桨的人工智能学习与实训社区,提供在线编程环境、免费GPU 算力、海量开源算法和开放数据,帮助开发者快速创建和部署模。初次使用的小伙伴记得注册之后完成新手礼包获取算力卡。
YOLO2COCO
因为小编拿到的数据集是yolo格式的,所以需要借助这个工具把YOLO格式的标签数据转成COCO格式数据集。
PaddleUtils
这是一个paddlepaddle模型减支工具,我们需要把训练得到的ppyoloe口罩检测模型进行裁剪。
paddleyolo
paddleyolo里面有很多目标检测算法,其中包括ppyoloe这个算法,使用的时候只需要配置一些文件就可以训练我们的模型了,非常方便。因为训练模型需要GPU,所以需要在AI Studio里面使用paddleyolo。
二、数据准备
数据来自一些公开的数据集,为了让模型训练得更好,自己也收集了一些数据,并使用labelimg对数据进行标注。
最后打包好数据,上传到AI Studio。
三、创建数据集
四、推理代码
from ope