在樹莓派上安裝 YOLOv8 需要安裝 Python、PyTorch、Ultralytics YOLOv8 以及相關的 OpenCV、NumPy 等庫。以下是完整的安裝步驟:
sudo apt update && sudo apt upgrade -y
sudo apt install python3-pip python3-dev python3-venv -y
樹莓派的 CPU 版本不同,需選擇適合的 PyTorch 版本。
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
pip install torch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 --no-cache-dir
pip install ultralytics
from ultralytics import YOLO
model = YOLO("yolov8n.pt")
print("YOLOv8 安裝成功!")
pip install opencv-python numpy
import cv2
print(cv2.__version__)
wget https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8n.pt
from ultralytics import YOLO
import cv2
model = YOLO("yolov8n.pt")
cap = cv2.VideoCapture(0)
while cap.isOpened():
ret, frame = cap.read()
if not ret:
break
results = model(frame)
annotated_frame = results[0].plot()
cv2.imshow("YOLOv8 Detection", annotated_frame)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
cap.release()
cv2.destroyAllWindows()
pip install openvino openvino-dev
然後使用 YOLO("yolov8n_openvino_model") 來執行。
pip install nvidia-pyindex
pip install nvidia-tensorrt
這樣就完成 YOLOv8 的安裝與運行!🎉 有問題可以隨時問我 😊