【yoloV3-keras】 keras-yolov3 进行批量测试并保存结果

墨蓝 2022-02-19 05:17 341阅读 0赞

几个月前自己上手YOLOV3-keras，自己训练了一个数据集。在测试的时候，发现源码作者的测试不好用。自己稍稍修改了一下。

来源于：[https://github.com/qqwweee/keras-yolo3][https_github.com_qqwweee_keras-yolo3]

# 几点改进 #

（1）批量测试图片

**将待测试的图片放入 './test'路径下。**

测试的时候，第一张图片需要的时间大约是 2s左右，因为需要加载模型，所需时间就相对较长一些。在博主的机器上，测试一一张图片的时间大约是0.1s左右；

（2）保存测试结果

完成测试后，将测试的结果保存在result文件夹中。方便以后查看

（3）将测试结果输出为一个result.txt文件。

result.txt内容包含了每一个bbox的信息。

# 效果 #

待检测的图片

![20190415192931507.png][]

检测结果

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3BsU29uZ19DU0RO_size_16_color_FFFFFF_t_70][]

result.txt中内容

> boy.jpg detect\_result：  
> find  3 target(s)   
> umbrella  score: 0.99444956   
> location: top: 195、 bottom: 379、 left: 152、 right: 412  
> person  score: 0.5912854   
> location: top: 1、 bottom: 62、 left: 396、 right: 442  
> person  score: 0.9996991   
> location: top: 227、 bottom: 549、 left: 237、 right: 379
> 
> cat.jpg detect\_result：  
> find  1 target(s)   
> cat  score: 0.988778   
> location: top: 48、 bottom: 729、 left: 281、 right: 769
> 
> dog.jpg detect\_result：  
> find  3 target(s)   
> dog  score: 0.98881614   
> location: top: 224、 bottom: 537、 left: 128、 right: 314  
> truck  score: 0.9140974   
> location: top: 85、 bottom: 170、 left: 475、 right: 688  
> bicycle  score: 0.9933555   
> location: top: 118、 bottom: 442、 left: 162、 right: 565
> 
> ourLaboratory.jpg detect\_result：  
> find  24 target(s)   
> mouse  score: 0.6115475   
> location: top: 978、 bottom: 1024、 left: 41、 right: 98  
> laptop  score: 0.34206218   
> .......  
> location: top: 544、 bottom: 721、 left: 65、 right: 176  
> person  score: 0.95439047   
> location: top: 616、 bottom: 771、 left: 194、 right: 398  
> person  score: 0.99081326   
> location: top: 418、 bottom: 580、 left: 680、 right: 780
> 
> time sum: 10.973031282424927s

四张图片一共用时约11s。

\---------------------------------------------------

有需要的程序可以在下面，先点赞、评论。

### Github：**[https://github.com/plsong/keras-yolo3-test][https_github.com_plsong_keras-yolo3-test]** ###

或者支持一下：[https://download.csdn.net/download/plsong\_csdn/11117636][https_download.csdn.net_download_plsong_csdn_11117636]

关于MAP的计算，大家可以参考我的另一篇文章！《[YOLOV3-keras版本的mAP计算][YOLOV3-keras_mAP]》

[https_github.com_qqwweee_keras-yolo3]: https://github.com/qqwweee/keras-yolo3
[20190415192931507.png]: /images/20220219/c83fa059c3d54bdcbc8458742c61b2e0.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3BsU29uZ19DU0RO_size_16_color_FFFFFF_t_70]: /images/20220219/fa0a7890f7b647b3a15a48d6648f0565.png
[https_github.com_plsong_keras-yolo3-test]: https://github.com/plsong/keras-yolo3-test
[https_download.csdn.net_download_plsong_csdn_11117636]: https://download.csdn.net/download/plsong_csdn/11117636
[YOLOV3-keras_mAP]: https://blog.csdn.net/plSong_CSDN/article/details/89502117