基于ai studio训练paddle-paddle模型

 一、创建项

1.在ai studio界面点击个人中心→项目→创建和Fork的项目→创建项目
2. 项目信息

①选择类型,Notebook

②配置环境,Notebook版本:BML Codelab;项目框架:PaddlePaddle2.x;项目环境不用选择

③项目描述,所填信息可随便选择;关于数据集,添加数据集所使用的是平台提供的数据集,创建数据集使用的是自己的数据集

3.页面信息

 ①点击+可以添加终端

②位于data文件夹下的内容不会被保存,除了平台提供的数据集

二、开始训练

1.训练环境准备
# 更换当前路径
cd /home/aistudio/work

# 下载PaddleDetection代码
git clone https://gitee.com/paddlepaddle/PaddleDetection

# 下载依赖
pip install -r PaddleDetection/requirements.txt

# 下载 pycocotools
pip insatll pycocotools
 2.数据集
# 解压数据
unzip -o /home/aistudio/data/data50329/HelmetDetection.zip -d PaddleDetection/dataset/ #第一个是数据集压缩包的路径,第二个是解压后的位置

 如果数据集不是voc格式的要转化为voc格式,voc格式可以把图片放在同一个文件夹,但是txt文件要在不同的文件夹

3.修改训练配置文件

①/home/aistudio/work/PaddleDetection/dataset/voc/label_list.txt   文件中的类要修改成我们数据集标注的所有类的集合

②/home/aistudio/work/PaddleDetection/configs/datasets/voc.yml  文件中的标签个数为3,dataset_dir为      ,anno_path修改为train.txt(我们之前创建的就是train.txt)

③/home/aistudio/work/PaddleDetection/configs/picodet/picodet_s_416_coco_lcnet.yml  

_BASE_修改为:../datasets/voc.yml  

扫描二维码关注公众号,回复: 16342891 查看本文章

pretrain_weights修改为: https://paddledet.bj.bcebos.com/models/picodet_s_416_coco_lcnet.pdparams(预训练权重)

4.训练模型
python PaddleDetection/tools/train.py -c PaddleDetection/configs/datasets/voc.yml

猜你喜欢

转载自blog.csdn.net/this_dudu/article/details/132189454