YOLOv5以txt或json格式輸出預(yù)測(cè)結(jié)果的方法詳解

更新時(shí)間：2023年03月14日 10:48:19 作者：lianlianerya

這篇文章主要給大家介紹了關(guān)于YOLOv5以txt或json格式輸出預(yù)測(cè)結(jié)果的相關(guān)資料,文中通過實(shí)例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下

1.YOLOv5源碼以多種格式輸出預(yù)測(cè)結(jié)果

1.run函數(shù)——傳入?yún)?shù)

2.run函數(shù)——保存打印

在對(duì)每張圖片做處理的循環(huán)里的寫入結(jié)果部分：

我們可以看到以下的代碼片段就是以不同格式輸出預(yù)測(cè)結(jié)果的代碼

2.YOLOv5以.txt格式輸出預(yù)測(cè)結(jié)果

1.執(zhí)行以下代碼就可以得到以.txt格式輸出預(yù)測(cè)結(jié)果

 python detect.py --weights runs/exp1/weights/best.pt --source inference/images/ --save-txt --save-conf

–save-txt save results to *.txt #將預(yù)測(cè)的bounding box保存為txt文件
–save-conf save confidences in --save-txt labels #類別的概率

每個(gè)txt會(huì)生成一行一個(gè)目標(biāo)的信息，信息包括類別序號(hào)、后面四個(gè)為bbox位置(xcenter ycenter w h)，最后一個(gè)是該類別的概率

2.輸出格式：

3.YOLOv5以.json格式輸出預(yù)測(cè)結(jié)果

1.需要在源碼中加上一段代碼第一步

在這個(gè)位置加入

save_json = True,# 輸出json文件save results to *.json
content_json = [],

第二步

在這個(gè)位置下面加上

 # 輸出 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.輸出格式

總結(jié)

到此這篇關(guān)于YOLOv5以txt或json格式輸出預(yù)測(cè)結(jié)果的文章就介紹到這了,更多相關(guān)YOLOv5 txt或json輸出預(yù)測(cè)結(jié)果內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章:

相關(guān)文章

淺談python中scipy.misc.logsumexp函數(shù)的運(yùn)用場(chǎng)景
下面小編就為大家?guī)硪黄獪\談python中scipy.misc.logsumexp函數(shù)的運(yùn)用場(chǎng)景。小編覺得挺不錯(cuò)的，現(xiàn)在就分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧
2016-06-06
python使用opencv resize圖像不進(jìn)行插值的操作
這篇文章主要介紹了python使用opencv resize圖像不進(jìn)行插值的操作，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
2020-07-07
PyTorch CNN實(shí)戰(zhàn)之MNIST手寫數(shù)字識(shí)別示例
本篇文章主要介紹了PyTorch CNN實(shí)戰(zhàn)之MNIST手寫數(shù)字識(shí)別示例，小編覺得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧
2018-05-05
Python之ReportLab繪制條形碼和二維碼的實(shí)例
下面小編就為大家分享一篇Python之ReportLab繪制條形碼和二維碼的實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
2018-01-01
Python使用win32 COM實(shí)現(xiàn)Excel的寫入與保存功能示例
這篇文章主要介紹了Python使用win32 COM實(shí)現(xiàn)Excel的寫入與保存功能,結(jié)合實(shí)例形式分析了Python調(diào)用win32 COM組件針對(duì)Excel文件的讀寫與保存相關(guān)操作技巧,需要的朋友可以參考下
2018-05-05
安裝python3.7編譯器后如何正確安裝opnecv的方法詳解
這篇文章主要介紹了安裝python3.7編譯器后如何正確安裝opnecv,本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下
2020-06-06
將Jupyter?Notebook(.ipynb)文件轉(zhuǎn)換為Python(.py)文件的3種方法
大多數(shù)數(shù)據(jù)科學(xué)在線課程都把Jupyter Notebook作為教學(xué)媒介,這是因?yàn)槌鯇W(xué)者在Jupyter Notebook的單元格中編寫代碼,比編寫包含類和函數(shù)的腳本更容易,這篇文章主要給大家介紹了關(guān)于將Jupyter?Notebook(.ipynb)文件轉(zhuǎn)換為Python(.py)文件的3種方法,需要的朋友可以參考下
2023-10-10
Python的SimpleHTTPServer模塊用處及使用方法簡介
這篇文章主要介紹了Python的SimpleHTTPServer模塊用處及使用方法簡介，小編覺得還是挺不錯(cuò)的，具有一定借鑒價(jià)值，需要的朋友可以參考下
2018-01-01
啟動(dòng)Atom并運(yùn)行python文件的步驟
在本篇文章中我們給大家分享了啟動(dòng)Atom并運(yùn)行python文件的步驟以及具體做法，需要的朋友們參考下。
2018-11-11
Python使用openpyxl讀寫excel文件的方法
本篇文章主要介紹了Python使用openpyxl讀寫excel文件的方法，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下
2017-06-06