1.YOLOv5源碼以多種格式輸出預測結果
1.run函數(shù)——傳入?yún)?shù)
2.run函數(shù)——保存打印
在對每張圖片做處理的循環(huán)里的寫入結果部分：
我們可以看到以下的代碼片段就是以不同格式輸出預測結果的代碼
2.YOLOv5以.txt格式輸出預測結果
1.執(zhí)行以下代碼就可以得到以.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 = [],
- 第二步
在這個位置下面加上
文章來源：http://www.zghlxwxcb.cn/news/detail-844016.html
# 輸出 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.輸出格式
文章來源地址http://www.zghlxwxcb.cn/news/detail-844016.html
到了這里，關于YOLOv5 以txt 或json格式輸出預測結果的文章就介紹完了。如果您還想了解更多內(nèi)容，請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關文章，希望大家以后多多支持TOY模板網(wǎng)！
![[yolov5] yolo的數(shù)據(jù)標簽格式](https://imgs.yssmx.com/Uploads/2024/02/442089-1.png)
