How to solve segmentation fault error on raspberry pi 4

Question

I am currently trying to use yolov8 to perform object detection on the raspberry pi 4. I have installed ultralytics and other necessary packages but whenever i run the code on the terminal it says "segmentation fault". Here is the code that I am trying to run:

from ultralytics import YOLO
model = YOLO("yolov8n.pt")
model.predict(source=0, show=True)

How can I fix this error?

Answer 1

Downgrading torch and torchvision worked for me.

pip uninstall torch torchvision
pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2

You can read more about this issue at https://github.com/ultralytics/ultralytics/issues/5274

Answer 2

Try this!

from ultralytics import YOLO

model = YOLO("yolov8n.pt")

results = model.predict(source=0, stream=True)

for r in results:
    boxes = r.boxes  # Bounding box coordinates
    masks = r.masks  # Segmentation masks
    probs = r.probs  # Class probabilities
r.render()