YOLOv5怎么以txt或json格式输出预测结果

这篇“YOLOv5怎么以txt或JSON格式输出预测结果”文章的知识点大部分人都不太理解，所以小编给大家总结了以下内容，内容详细，步骤清晰，具有一定的借鉴价值，希望大家阅读完这篇文章能有所收获，下面我们一起来看看这篇“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 *.jsoncontent_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格式输出预测结果”这篇文章的内容，相信大家都有了一定的了解，希望小编分享的内容对大家有帮助，若想了解更多相关的知识内容，请关注编程网精选频道。