Window10的Pytorch+Yolov5的安裝與使用

筆者使用Window10，Nvidia GTX 1660Ti顯卡。

Step1)安裝YOLOV5(https://github.com/ultralytics/yolov5)

#cd c:\

#git clone https://github.com/ultralytics/yolov5

Step2)安裝Python虛擬環境

#conda create -n yolov5 python=3.8

#conda activate yolov5

Step4)安裝pytorch(https://pytorch.org/get-started/locally/)，雖然pytorch網站已經說不支援cuda10.2，但是透過以下指令安裝還是可行。

#conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

Step5)安裝yolov5所需套件

#cd c:\yolov5

#pip install -r requirement.txt

Step6)開啟攝影鏡頭進行辨識，左邊為攝影機的輸出結果，右邊為即時的辨識結果

#python detect.py --source 0

Step8)使用Youtube影片進行辨識，會自動安裝pafy與youtube_dl。

#python detect.py --source https://www.youtube.com/watch?v=xxxxxxxxx