YOLOv5以txt或json格式輸出預測結果的方法詳解
更新時間：2023年03月14日 10:48:19 作者：lianlianerya
這篇文章主要給大家介紹了關于YOLOv5以txt或json格式輸出預測結果的相關資料,文中通過實例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下
1.YOLOv5源碼以多種格式輸出預測結果
1.run函數(shù)——傳入參數(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 = [],
第二步
在這個位置下面加上
# 輸出 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輸出預測結果內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！
相關文章
淺談python中scipy.misc.logsumexp函數(shù)的運用場景
下面小編就為大家?guī)硪黄獪\談python中scipy.misc.logsumexp函數(shù)的運用場景。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2016-06-06
python使用opencv resize圖像不進行插值的操作
這篇文章主要介紹了python使用opencv resize圖像不進行插值的操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-07-07
PyTorch CNN實戰(zhàn)之MNIST手寫數(shù)字識別示例
本篇文章主要介紹了PyTorch CNN實戰(zhàn)之MNIST手寫數(shù)字識別示例，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-05-05
下面小編就為大家分享一篇Python之ReportLab繪制條形碼和二維碼的實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-01-01
Python使用win32 COM實現(xiàn)Excel的寫入與保存功能示例
這篇文章主要介紹了Python使用win32 COM實現(xiàn)Excel的寫入與保存功能,結合實例形式分析了Python調用win32 COM組件針對Excel文件的讀寫與保存相關操作技巧,需要的朋友可以參考下2018-05-05
安裝python3.7編譯器后如何正確安裝opnecv的方法詳解
這篇文章主要介紹了安裝python3.7編譯器后如何正確安裝opnecv,本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2020-06-06
將Jupyter?Notebook(.ipynb)文件轉換為Python(.py)文件的3種方法
大多數(shù)數(shù)據科學在線課程都把Jupyter Notebook作為教學媒介,這是因為初學者在Jupyter Notebook的單元格中編寫代碼,比編寫包含類和函數(shù)的腳本更容易,這篇文章主要給大家介紹了關于將Jupyter?Notebook(.ipynb)文件轉換為Python(.py)文件的3種方法,需要的朋友可以參考下2023-10-10
Python的SimpleHTTPServer模塊用處及使用方法簡介
這篇文章主要介紹了Python的SimpleHTTPServer模塊用處及使用方法簡介，小編覺得還是挺不錯的，具有一定借鑒價值，需要的朋友可以參考下2018-01-01
在本篇文章中我們給大家分享了啟動Atom并運行python文件的步驟以及具體做法，需要的朋友們參考下。2018-11-11
本篇文章主要介紹了Python使用openpyxl讀寫excel文件的方法，具有一定的參考價值，感興趣的小伙伴們可以參考一下2017-06-06