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

本文介绍了如何使用Keras-YOLOv3进行批量图片测试,并保存检测结果。通过改进代码,实现了快速批量处理图片,每个测试大约0.1s,结果保存为result.txt,包含每个边界框的详细信息。测试实例展示在四张图片上的检测效果,总耗时约11s。完整代码可在Github:https://github.com/plsong/keras-yolo3-test 查看。
摘要由CSDN通过智能技术生成

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

来源于:https://github.com/qqwweee/keras-yolo3

几点改进

(1)批量测试图片

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

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

(2)保存测试结果

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

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

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

效果

待检测的图片

检测结果

result.txt中内容

要使用yolov3-keras训练自己的数据集,您需要按照以下步骤进行操作: 1. 数据集准备: 首先,您需要准备您自己的数据集,并将其进行标注。您可以使用VOC2007数据集作为参考。确保您的数据集包含正确的文件夹结构,并且您的标注文件与图像文件匹配。 2. 生成训练集和验证集: 根据您的数据集,创建一个用于训练的文件夹和一个用于验证的文件夹。将您的数据集按比例划分到这两个文件夹中。 3. 修改参数文件: 根据您的数据集和需求,修改yolov3.cfg参数文件。确保您根据您的类别数量,修改参数文件中的类别数相关的选项。同时,您还可以根据需要调整其他参数,如输入图片大小等。 4. 修改类别文件: 创建一个voc_classes.txt文件,并在其中列出您的数据集中的类别名称。确保类别名称的顺序与您的标注文件中的类别顺序一致。 5. 修改训练脚本: 修改train.py文件,以便加载您的数据集和参数文件。确保您修改了相关的路径和文件名,并设置正确的训练参数,如批量大小、学习率等。 6. 开始训练: 运行train.py文件,开始训练您的模型。此时,您需要等待训练过程完成。训练时间可能会根据数据集的大小和计算资源的性能而有所不同。 请注意,您还需要安装相应的环境依赖,如tensorflow-gpu和keras,确保您的环境与yolov3-keras的要求一致。另外,您还可以参考相关的行动计划和具体实现内容,以了解更多关于训练自己数据集的细节。 总结起来,要使用yolov3-keras训练自己的数据集,您需要准备数据集、生成训练集和验证集、修改参数文件、修改类别文件、修改训练脚本并开始训练。记得根据自己的需求进行相应的调整和修改。希望这些信息对您有所帮助!
评论 134
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值