YOLOv5以txt或json格式输出预测结果的方法详解
更新时间：2023年03月14日 10:48:19 作者：lianlianerya
这篇文章主要给大家介绍了关于YOLOv5以txt或json格式输出预测结果的相关资料,文中通过实例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
1.YOLOv5源码以多种格式输出预测结果
1.run函数——传入参数
2.run函数——保存打印
在对每张图片做处理的循环里的写入结果部分：
我们可以看到以下的代码片段就是以不同格式输出预测结果的代码
2.YOLOv5以.txt格式输出预测结果
1.执行以下代码就可以得到以.txt格式输出预测结果
python detect.py --weights runs/exp1/weights/best.pt --source inference/images/ --save-txt --save-conf
–save-txt save results to *.txt #将预测的bounding box保存为txt文件
–save-conf save confidences in --save-txt labels #类别的概率
每个txt会生成一行一个目标的信息，信息包括类别序号、后面四个为bbox位置(xcenter ycenter w h)，最后一个是该类别的概率
2.输出格式：
3.YOLOv5以.json格式输出预测结果
1.需要在源码中加上一段代码 第一步
在这个位置加入
save_json = True,# 输出json文件save results to *.json content_json = [],
第二步
在这个位置下面加上
# 输出 json 文件
if save_json:
# windows下使用
num += 1
file_name = save_path.split('\\')
# Linux下使用
# file_name = save_path.split('/')
content_dic = {
"name": file_name[len(file_name) - 1],
"num": num,
"category": (names[int(cls)]),
"bbox": torch.tensor(xyxy).view(1, 4).view(-1).tolist(),
"score": conf.tolist()
}
content_json.append(content_dic)
2.输出格式
总结
到此这篇关于YOLOv5以txt或json格式输出预测结果的文章就介绍到这了,更多相关YOLOv5 txt或json输出预测结果内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要介绍了关于最大池化层和平均池化层图解，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-12-12
今天小编就为大家分享一篇基于Python中isfile函数和isdir函数使用详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-11-11
这篇文章主要介绍了python 模拟登陆163邮箱的示例，帮助大家更好的理解和使用python，感兴趣的朋友可以了解下2020-12-12
这篇文章主要给大家介绍了关于python创建与遍历二叉树的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2021-03-03
这篇文章主要介绍了如何基于Python爬取隐秘的角落评论,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-07-07
这篇文章主要介绍了pyqt4实现半透明的天气预报界面示例,需要的朋友可以参考下2014-03-03
Matplotlib是一个Python的2D绘图库,通过Matplotlib开发者可以仅需要几行代码,便可以生成绘图,直方图,功率谱,条形图,错误图,散点图等,这篇文章主要给大家介绍了Python中如何使用Matplotlib库绘制图形的相关资料2022-07-07
本文主要介绍了Pandas中inf值替换的方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2022-08-08
fixture是pytest特有的功能，用以在测试执行前和执行后进行必要的准备和清理工作，这篇文章主要介绍了pytest中的fixture基本用法,需要的朋友可以参考下2023-02-02
大家好，本篇文章主要讲的是python简单批量梯度下降代码，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下，方便下次浏览2022-01-01
最新评论