本文介绍了基于PaddleDetection的目标检测项目流程，包括克隆仓库、安装依赖、编译安装等步骤。还涵盖数据集准备，如查看COCO标注、解压和配置数据集，以及模型训练、预测、导出，最后说明代码提交及优化方法，如调整阈值等。

一、安装PaddleDetection

将PaddleDetection/configs/picodet/picodet_m_320_coco_lcnet.yml里的数据集配置改成我们前面自己新建的数据集配置，也就是coco_dog.yml

2、模型训练

注意:数据集配置一定要改为前面我们自行配置的

# 模型训练%cd ~%cd PaddleDetection!python tools/train.py -c configs/picodet/picodet_m_320_coco_lcnet.yml登录后复制 ? ?

如果需要更换模型，直接在上述代码中修改路径即可，注意：更换模型时，检查数据集是不是我们要用的数据集。

如果训练时，突然中断了，需要恢复训练，使用-r参数，后面加上模型路径即可

例：!python tools/train.py -c configs/picodet/picodet_m_320_coco_lcnet.yml -r -r output/picodet_m_320_coco_lcnet/best_model

想要边训练边评估，末尾加 --eval 即可

例：!python tools/train.py -c configs/picodet/picodet_m_320_coco_lcnet.yml --eval

训练时的日志输出已单独保存在vdl_log_dir/scalar/vdlrecords.1649487648.log

注意：以下代码运行，需等模型训练完成后

3.模型预测

训练好模型以后，我们可以检验一下效果。

%cd ~%cd PaddleDetection# 更换”--infer_img“里的图片路径以预测不同的图片!python tools/infer.py -c configs/picodet/picodet_m_320_coco_lcnet.yml --infer_img=/home/aistudio/WisdomGuide/val/no_light_629.png --output_dir=infer_output/ --draw_threshold=0.5 -o weights=/home/aistudio/PaddleDetection/output/picodet_m_320_coco_lcnet/best_model.pdparams --use_vdl=Ture登录后复制 ? ?

4.导出模型

转换并保存静态图模型。

这里假设我们已经在线训练10小时，获得了模型

如果要生成版本，由于PaddleDetection文件夹下的文件较多，无法保存，故要把训练好的模型单独拿出来，例如放在/home/aistudio/checkpoint/best_model.pdparams。

%cd ~%cd PaddleDetection# 将”-o weights“里的模型路径换成你自己训好的模型!export CUDA_VISIBLE_DEVICES=0!python tools/export_model.py -c configs/picodet/picodet_m_320_coco_lcnet.yml -o weights=/home/aistudio/PaddleDetection/output/picodet_m_320_coco_lcnet/best_model.pdparams TestReader.fuse_normalize=true登录后复制 ? ?

四、代码提交

这一步与Baseline操作基本一样

AI Studio上提供了一套测评系统，我们将训练好的模型和预测代码打包上传，测评系统会自动算分。

提交格式：

| -- model| | -- xxx.pb| | … | -- env| | -- | |…| -- predict.py| -- …登录后复制 ? ?In [5]

%cd ~# 在work目录下整理提交代码# 创建model文件夹（已经创建好的话，就不用再创建了）!mkdir /home/aistudio/work/model# 将检测模型拷贝到model文件夹中!cp -r /home/aistudio/PaddleDetection/output_inference/picodet_m_320_coco_lcnet/ /home/aistudio/work/model/# 将训练脚本拷贝到work目录下，与model同级!cp /home/aistudio/PaddleDetection/tools/train.py /home/aistudio/work/# 将PaddleDetection也复制进来!cp -r /home/aistudio/PaddleDetection/ /home/aistudio/work/登录后复制 ? ? ? ?

/home/aistudio登录后复制 ? ? ? ?

这里为了让压缩包更小，建议把PaddleDetection里面不用的模型备份出来。特别地，隐藏文件.git占用存储空间较大，且在这里没有什么用，故可以手动删除。

# 删除隐藏文件!rm -rf /home/aistudio/work/PaddleDetection/.git/登录后复制 ? ?

注意： 在打包前，请将work/PaddleDetection/deploy/python/preprocess.py的第17行改为：

from .keypoint_preprocess import get_affine_transform登录后复制 ? ? ? ?

否则会出现以下报错：

Traceback (most recent call last): File ”predict.py“, line 19, in <module> from deploy.python.preprocess import preprocess, Resize, NormalizeImage, Permute, PadStride File ”PaddleDetection/deploy/python/preprocess.py“, line 17, in <module> from keypoint_preprocess import get_affine_transformModuleNotFoundError: No module named 'keypoint_preprocess'登录后复制 ? ?In [10]

# 打包代码%cd /home/aistudio/work/!zip -r -q -o submission.zip model/ PaddleDetection/ train.py predict.py登录后复制 ? ? ? ?

/home/aistudio/work登录后复制 ? ? ? ?

在提交前，我们需要改动一下predict.py文件，这样我们才能上0.8?

我们把阈值参数由0.05改为0.5，就OK啦。

预选赛时，有不少同学说：我模型的预测效果比Baseline好，为啥提交后分数比Baseline低啊，原因就是这个。

阈值由Baselin的0.05改为0.5，我们就会获得巨大提升。

然后从aistudio上下载打包好的submission.zip文件，就可以直接到官网提交啦！现在预选赛的通道关闭了，我们可以在练习赛进行提交。

如果提交成功，就可以看到自己的分数了。

本项目在baseline基础上，将阈值改为0.5，结果在0.8之上。大家可以试试。

五、优化方法

1、尝试不同的阈值，找出最佳阈值。

修改阈值能够提升一点我们的得分，0.5对有些模型来说，不是最佳阈值，我们可以试试不同阈值提交，找到该模型的最佳阈值。

2、加大训练轮数

有些模型原始的配置轮数不是最大，我们可以将训练轮数加大到，该模型最大轮数，例如Picodet-m-416模型

最大轮数是300轮，最初设置是250轮，我们可以将轮数设置最大轮数300轮。

3、尝试不同的模型

尝试一下其他的模型，例如PP-YOLO，YOLOv3等，这里透露一下，PP-Picodet能上0.87

4、加载COCO数据集预训练模型，使用预训练模型可以有效提升模型精度

以Picodet-m-320为例： 修改Picodet_v2.yml中的pretrain_weights即可?

5、修改loss

将picodet_v2.yml中的lossbbox name值由GIoU改为DIou?

6、修改lr

降低学习率，尝试调整为一半或十分之一，亦或是其他学习率

以上就是一些通用的方法，非常基础外加非常简单，大家可以逐一尝试，或者觉得麻烦的话，一口气全部试在一个模型里面，模型训练完成后，通过智慧导盲练习赛进行得分验证。此外这些方法也可以在以后的一些比赛使用，屡试不爽。

或许还有一些其他不动网络的提升方法，但我并不知道，欢迎各位大佬补充

我在work目录下面放了一个导出的Picodet-m-320模型,大家把predict.py的阈值改改，复制PaddleDetion，打包进行提交，验证第一个方法。