从零开始使用Detectron训练第三方数据集是什么体验（六）

短命女 2022-03-20 17:10 152阅读 0赞

**从零开始使用Detectron训练第三方数据集是什么体验（六）**

> **本栏目分为6个部分：**
> 
>  *  [第1部分：Detectron框架简介与安装][1_Detectron]
>  *  [第2部分（本文）：使用LabelImgPlus对图片进行标注][2_LabelImgPlus]
>  *  [第3部分：数据集处理][3]
>  *  [第4部分：在detectron上部署数据集][4_detectron]
>  *  [第5部分：训练][5]
>  *  第6部分（本文）：测试  
>     GitHub： [https://github.com/zhiweichen95/detectron\_my][https_github.com_zhiweichen95_detectron_my]

**第6部分：测试**

### 文章目录 ###

*  0 前言
 *  1 复现测试数据集的mAP计算
 *  2 使用一些图片进行测试并可视化

# 0 前言 #

训练完成后，自动会进行测试输出结果。如果还需要复现测试结果，本章节内容将详述。

# 1 复现测试数据集的mAP计算 #

python ./tools/test.py --cfg configs/my/retinanet_R-50-FPN_1x1.0.yaml \
                           TEST.WEIGHTS experiments/2.0/train/voc_2007_train/retinanet/model_final.pkl

*  `--cfg` ：配置文件路径
 *  `TEST.WEIGHTS`：进行测试模型的路径

# 2 使用一些图片进行测试并可视化 #

本项目需要测试的图片jpg存放的文件夹路径为：`$Detectron/mytest/input/`

python ./tools/infer_simple.py --cfg configs/my/retinanet_R-50-FPN_1x1.0.yaml \
                                   --wts experiments/2.0/train/voc_2007_train/retinanet/model_final.pkl \
                                   --output-dir mytest/output/ \
                                   mytest/input/

*  `--cfg` ：配置文件路径
 *  `--wts`：进行测试模型的路径
 *  `--output-dir`：可视化图片的保存文件夹路径
 *  `mytest/input/`指的是需要测试的图片jpg存放的文件夹路径

这样就可以进行目标检测辣~

可以看看本人跑出的结果：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3p3X19jaGVu_size_16_color_FFFFFF_t_70]  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3p3X19jaGVu_size_16_color_FFFFFF_t_70 1]

--------------------

本项目就到此结束啦。如有任何问题，欢迎留言交流学习。

[1_Detectron]: https://blog.csdn.net/zw__chen/article/details/86723277
[2_LabelImgPlus]: https://blog.csdn.net/zw__chen/article/details/87188004
[3]: https://blog.csdn.net/zw__chen/article/details/87202034
[4_detectron]: https://blog.csdn.net/zw__chen/article/details/87209669
[5]: https://blog.csdn.net/zw__chen/article/details/87210932
[https_github.com_zhiweichen95_detectron_my]: https://github.com/zhiweichen95/detectron_my
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3p3X19jaGVu_size_16_color_FFFFFF_t_70]: /images/20220320/01fa76a466f44613937c9fd6bc376601.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3p3X19jaGVu_size_16_color_FFFFFF_t_70 1]: /images/20220320/533189c60be8412f906a1e187cda8586.png