yolo是一個object detect model,顧名思義就是拿來檢測物件的。
使用yolo官網的安裝步驟
git clone https://github.com/ultralytics/yolov5 # clone
cd yolov5
pip install -r requirements.txt # install
在全新的環境中去執行的話,可能最終安裝完成之後會變成只能夠使用CPU來運算的方式。
我解決的方式是,先將本地安裝好的pytorch刪除,然後去pytorch官網https://pytorch.org/get-started/locally/選擇適用於自己的選項來安裝。
然後再根據yolo官網的安裝步驟安裝環境,完成後,即會使用GPU來運算。速度會比CPU快得多。