Yolov8的復現流程主要包含環境配置、下載源碼和驗證環境三大步驟:
-
環境配置
- 查看電腦狀況:通過任務管理器查看電腦是否有獨立顯卡(NVIDIA卡)。若有,后續可安裝GPU版本的pytorch以加速訓練;若沒有,則安裝CPU版本,不過訓練速度會較慢。
- 安裝所需軟件
- Anaconda3安裝:可從清華鏡像源選擇對應系統版本的Anaconda3安裝包下載。安裝時選擇“Just Me”,安裝路徑建議為非C盤的純英文路徑,如“D:\Anaconda3” 。安裝過程中記得勾選添加到環境變量的選項,推薦全選相關配置選項。
- Pycharm安裝:在官網下載社區免費版,下載后打開安裝程序,選擇安裝路徑(建議非C盤) ,安裝選項中勾選創建桌面快捷方式、更新PATH變量等,然后點擊安裝。
- 安裝環境
-
安裝cuda及cudnn:安裝pytorch前需安裝cuda。通過“nvidia - smi”命令查看顯卡支持的CUDA最高版本,若版本不足需更新顯卡驅動。在NVIDIA Developer官網或提供的網盤鏈接下載匹配版本的cuda,安裝時建議選默認路徑,安裝過程中選擇自定義安裝并全選組件。安裝完成后用“nvcc - V”命令檢查是否安裝成功。之后,進入cudnn官網,下載與cuda版本匹配的cudnn Windows版本壓縮包,解壓后將里面的三個文件夾內容復制到cuda的安裝目錄下進行覆蓋替換。
-
創建虛擬環境:打開Anaconda Prompt,修改pip源和conda源為中科大源以加速下載。然后創建名為yolov8、python版本為3.9的虛擬環境(若使用教程指定的pytorch源及版本,python版本需為3.9),命令為“conda create -n yolov8 python = 3.9” ,創建完成后用“conda activate yolov8”命令進入該環境。
-
安裝GPU版本(有NVIDIA顯卡):在yolov8環境中,使用命令“pip install torch2.0.0+cu118 torchvision0.15.1+cu118 --extra - index - url https://download.pytorch.org/whl/cu118”安裝pytorch的GPU版本。若下載失敗,可從夸克網盤下載對應版本的torch文件,通過“pip install”安裝本地文件的方式安裝。安裝完成后,使用“pip install ultralytics”命令安裝ultralytics庫,若安裝報錯,可使用“pip install ultralytics -i https://pypi.tuna.tsinghua.edu.cn/simple”命令安裝。
-
-
安裝CPU版本(無NVIDIA顯卡):在yolov8環境中,使用命令“pip install torch2.0.0 torchvision0.15.1 torchaudio==2.0.1 --index - url https://download.pytorch.org/whl/cpu”安裝pytorch的CPU版本。安裝完成后,同樣使用“pip install ultralytics”命令安裝ultralytics庫,報錯處理方式與GPU版本安裝時相同。
-
-
下載yolov8源碼:可從GitHub倉庫“https://github.com/ultralytics/ultralytics/”下載yolov8源碼,也可從夸克網盤下載。
-
驗證環境:將下載的預訓練權重文件復制到ultralytics根目錄(如ultralytics - 8.2.0目錄)下,從網上找一張圖片(如貓貓圖片),命名為cat.jpg并放入相同目錄。在yolov8環境下,通過命令行運行“yolo predict model=yolov8n.pt source=‘cat.jpg’” ,或者在Pycharm中新建yolov8_predict.py文件,添加如下代碼運行驗證:
from ultralytics import YOLO
# 加載預訓練的 YOLOv8n 模型
model = YOLO('yolov8n.pt')
# 定義圖像文件的路徑
source = 'cat.jpg'
# 運行推理,并附加參數
model.predict(source, save=True)
運行后若出現類似 “image 1/1… 1 cat, …ms” 且結果保存路徑正確的信息,即表明環境正常。若報錯,根據報錯信息進行相應處理,如 “ModuleNotFoundError: No module named ‘hub_sdk‘” 錯誤,執行 “pip install hub_sdk” ;“ImportError: DLL load failed while importing _imaging: 找不到指定的模塊” 錯誤,執行 “pip install pillow8.4.0” ;numpy相關錯誤,執行 “pip install numpy1.26.3” 等。