几个月前自己上手YOLOV3-keras,自己训练了一个数据集。在测试的时候,发现源码作者的测试不好用。自己稍稍修改了一下。
几点改进
(1)批量测试图片
将待测试的图片放入 './test'路径下。
测试的时候,第一张图片需要的时间大约是 2s左右,因为需要加载模型,所需时间就相对较长一些。在博主的机器上,测试一一张图片的时间大约是0.1s左右;
(2)保存测试结果
完成测试后,将测试的结果保存在result文件夹中。方便以后查看
(3)将测试结果输出为一个result.txt文件。
result.txt内容包含了每一个bbox的信息。
2019-12-21 20:27:17
3KB
深度学习
1