如何在自定義的數(shù)據(jù)集上訓練YOLOv5?
在我們的自定義數(shù)據(jù)集上可視化tensorboard結(jié)果如果你因為一些原因不能把張量可視化,可以用utils.plot_result來繪制并保存為result.png。

你需要在驗證評估分數(shù)達到其最高點處獲取訓練好的模型權(quán)重。可視化YOLOv5訓練數(shù)據(jù)在訓練過程中,我們可以可視化真實訓練數(shù)據(jù)和增強后的訓練數(shù)據(jù)。

我們的真實訓練數(shù)據(jù)

我們的訓練數(shù)據(jù)采用自動YOLOv5增強對測試圖像運行YOLOv5推斷現(xiàn)在我們利用我們訓練好的模型,對測試圖像進行推理。訓練完成后,模型權(quán)重將保存到 weights/。推理過程,我們調(diào)用這些權(quán)重和一個指定模型置信度的conf(要求的置信度越高,預(yù)測越少)、以及一個推理源。源可以接受一個包含圖像、單個圖像、視頻文件以及設(shè)備的網(wǎng)絡(luò)攝像頭端口的目錄。對于源代碼,我將testjpg移到test-infer/。!python detect.py --weights weights/last_yolov5s_custom.pt --img 416 --conf 0.4 --source ../test_infer推理時間非常快,在我們的 Tesla P100 上,YOLOv5s 達到了每秒142幀!!

以142 FPS(0.007s/圖像)的速度推斷YOLOv5s最后,我們在測試圖像上可視化我們的檢測器推斷結(jié)果。

測試圖像的YOLOv5推理導出并保存YOLOv5權(quán)重以供將來推斷既然我們定制的YOLOv5物體檢測器已經(jīng)通過驗證,我們需要從Colab中取出權(quán)重,用于實時計算機視覺任務(wù),為此我們導入一個Google驅(qū)動器模塊并將其發(fā)送出去。from google.colab import drivedrive.mount('/content/gdrive')
%cp /content/yolov5/weights/last_yolov5s_custom.pt /content/gdrive/My Drive結(jié)論我們希望你可以訓練屬于你自己的定制YOLOv5檢測器!使用 YOLOv5 非常方便,而且訓練迅速,推理迅速,表現(xiàn)出色。讓我們把它弄出來!參考鏈接:https://blog.roboflow.a(chǎn)i/how-to-train-yolov5-on-a-custom-dataset/
請輸入評論內(nèi)容...
請輸入評論/評論長度6~500個字
最新活動更多
-
即日-5.20立即下載>> 【限時免費】物理場仿真助力生物醫(yī)學領(lǐng)域技術(shù)創(chuàng)新
-
精彩回顧立即查看>> 【直播】 智測未來·2026海克斯康春季產(chǎn)品創(chuàng)新日
-
精彩回顧立即查看>> 【線下論壇】新唐科技×芯唐南京 2026 年度研討會
-
精彩回顧立即查看>> OFweek 2026(第十五屆)中國機器人產(chǎn)業(yè)大會
-
精彩回顧立即查看>> 維科杯· OFweek 2025中國機器人行業(yè)年度評選
-
精彩回顧立即查看>> 【在線會議】液冷服務(wù)器信號完整性及冷卻液關(guān)鍵電參數(shù)測試
推薦專題
- 1 AI狂歡遇上油價破百,全球股市還能漲多久? | 產(chǎn)聯(lián)看全球
- 2 OpenAI深夜王炸!ChatGPT Images 2.0實測:中文穩(wěn)、細節(jié)炸,設(shè)計師慌了
- 3 6000億美元估值錨定:字節(jié)跳動的“去單一化”突圍與估值重構(gòu)
- 4 Tesla AI5芯片最新進展總結(jié)
- 5 連夜測了一波DeepSeek-V4,我發(fā)現(xiàn)它可能只剩“審美”這個短板了
- 6 熱點丨AI“瑜亮之爭”:既生OpenClaw,何生Hermes?
- 7 AI界的殺豬盤:9秒刪庫跑路,全員被封號,還繼續(xù)扣錢!
- 8 2026,人形機器人只贏了面子
- 9 DeepSeek降價90%:價格屠夫不是身份,是戰(zhàn)略
- 10 AI Infra產(chǎn)業(yè)鏈卡在哪里了?
- 高級軟件工程師 廣東省/深圳市
- 自動化高級工程師 廣東省/深圳市
- 光器件研發(fā)工程師 福建省/福州市
- 銷售總監(jiān)(光器件) 北京市/海淀區(qū)
- 激光器高級銷售經(jīng)理 上海市/虹口區(qū)
- 光器件物理工程師 北京市/海淀區(qū)
- 激光研發(fā)工程師 北京市/昌平區(qū)
- 技術(shù)專家 廣東省/江門市
- 封裝工程師 北京市/海淀區(qū)
- 結(jié)構(gòu)工程師 廣東省/深圳市


分享













