YOLOv5以txt或json格式輸出預(yù)測(cè)結(jié)果的方法詳解
更新時(shí)間：2023年03月14日 10:48:19 作者：lianlianerya
這篇文章主要給大家介紹了關(guān)于YOLOv5以txt或json格式輸出預(yù)測(cè)結(jié)果的相關(guān)資料,文中通過(guò)實(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)里的寫(xiě)入結(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 #類(lèi)別的概率
每個(gè)txt會(huì)生成一行一個(gè)目標(biāo)的信息，信息包括類(lèi)別序號(hào)、后面四個(gè)為bbox位置(xcenter ycenter w h)，最后一個(gè)是該類(lèi)別的概率
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)文章
這篇文章主要介紹了關(guān)于最大池化層和平均池化層圖解，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2022-12-12
基于Python中isfile函數(shù)和isdir函數(shù)使用詳解
今天小編就為大家分享一篇基于Python中isfile函數(shù)和isdir函數(shù)使用詳解，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-11-11
這篇文章主要介紹了python 模擬登陸163郵箱的示例，幫助大家更好的理解和使用python，感興趣的朋友可以了解下2020-12-12
python創(chuàng)建與遍歷二叉樹(shù)的方法實(shí)例
這篇文章主要給大家介紹了關(guān)于python創(chuàng)建與遍歷二叉樹(shù)的相關(guān)資料，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2021-03-03
這篇文章主要介紹了如何基于Python爬取隱秘的角落評(píng)論,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-07-07
pyqt4教程之實(shí)現(xiàn)半透明的天氣預(yù)報(bào)界面示例
這篇文章主要介紹了pyqt4實(shí)現(xiàn)半透明的天氣預(yù)報(bào)界面示例,需要的朋友可以參考下2014-03-03
Python中如何使用Matplotlib庫(kù)繪制圖形
Matplotlib是一個(gè)Python的2D繪圖庫(kù),通過(guò)Matplotlib開(kāi)發(fā)者可以?xún)H需要幾行代碼,便可以生成繪圖,直方圖,功率譜,條形圖,錯(cuò)誤圖,散點(diǎn)圖等,這篇文章主要給大家介紹了Python中如何使用Matplotlib庫(kù)繪制圖形的相關(guān)資料2022-07-07
本文主要介紹了Pandas中inf值替換的方法，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2022-08-08
fixture是pytest特有的功能，用以在測(cè)試執(zhí)行前和執(zhí)行后進(jìn)行必要的準(zhǔn)備和清理工作，這篇文章主要介紹了pytest中的fixture基本用法,需要的朋友可以參考下2023-02-02
大家好，本篇文章主要講的是python簡(jiǎn)單批量梯度下降代碼，感興趣的同學(xué)趕快來(lái)看一看吧，對(duì)你有幫助的話(huà)記得收藏一下，方便下次瀏覽2022-01-01