安裝YOLOv5模型環境並測試

聒噪的珍珠丸子發表於2020-11-21

安裝YOLOv5模型環境並測試

下載yolov5原始碼

YOLOv5 模型原始碼:https://github.com/ultralytics/yolov5

配置環境

下載後可以在資料夾中看到requirements.txt檔案,其中所給的就是YOLOv5模型所需要的環境配置。因為跑不同的模型所需要的環境可能不同,在此建議大家可以使用虛擬環境來進行配置。
詳細步驟見我之前的部落格:win10下載Anaconda並建立虛擬環境安裝pytorch
設定好虛擬環境後(或者你就在本機)開啟Anaconda Prompt,如果設定了虛擬環境就先activate 你的環境名進入。
使用 cd 進入到YOLOv5資料夾,來到requirements.txt 所在的路徑下,輸入

pip install -U -r requirements.txt

大部分的環境用這條語句都是可以配置上的,大概會卡在torch的下載上,大家同樣可以參考我之前的那篇部落格win10下載Anaconda並建立虛擬環境安裝pytorch,自己手動安裝torch和torchvision。

測試環境

如果環境都安裝完成了,下面我們測試一下YOLOv5的模型是否可以成功執行,開啟spyder(如果你設定了虛擬環境,就開啟帶有你虛擬環境名字尾的spyder)。執行detect.py檔案。這是在用他本來跑好的模型來進行測試,測試的圖片在yolov5/inference/images裡面,可以自己手動加,這裡我就自己加了幾張圖片,原本的檔案中應該只有自帶的兩張圖片。如果可以看到如下的情況,並在yolov5/inference/output資料夾中看到標註好的檔案,那麼恭喜你,你的YOLOv5模型環境配置完成啦!
在這裡插入圖片描述
參考:
https://blog.csdn.net/laovife/article/details/106802725?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-2.channel_param&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-2.channel_param

https://blog.csdn.net/ai_faker/article/details/107099907

最後附上我在安裝過程中幫助比較大的兩篇文章,我在安裝的過程中遇到了很多的問題,在此沒有一一列舉出來,如果大家遇到了問題歡迎給我留言交流,希望這篇文章可以幫助到大家!後續我會持續更新使用YOLOv5模型跑自己的資料集以及引數的調整。

相關文章