深度学习100例 | 第53天:用YOLOv5训练自己的数据集(超级详细完整版)
生活随笔
收集整理的這篇文章主要介紹了
深度学习100例 | 第53天:用YOLOv5训练自己的数据集(超级详细完整版)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
大家好,我是K同學啊!
我們接著上一篇文章 深度學習100例 | 第51天-目標檢測算法(YOLOv5)(入門篇) 配置完YOLOv5需要的環境后,今天我們試著用YOLOv5訓練自己的數據。(在開始本教程前,記得先跑一遍入門篇,確保其他環境是正常的)
有圖有真相,先看看我昨天的運行結果
【YOLOv5 源碼地址】
?? 我的環境:
- 語言環境:Python3.8
- 編譯器:PyCharm
- 深度學習環境:
- torch==1.10.0+cu113
- torchvision==0.11.1+cu113
- 顯卡:GeForce RTX 3080
文章目錄
- 一、準備好自己的數據
- 二、運行 split_train_val.py 文件
- 三、生成 train.txt、test.txt、val.txt 文件
總結
以上是生活随笔為你收集整理的深度学习100例 | 第53天:用YOLOv5训练自己的数据集(超级详细完整版)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 小白入门PyTorch | 第一篇:什么
- 下一篇: 第二篇:自动微分