Skip to content

Latest commit

 

History

History

pytorch_yolov8

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
cv_viewer
cv_viewer
 
 
ogl_viewer
ogl_viewer
 
 
README.md
README.md
 
 
detector.py
detector.py
 
 

README.md

ZED SDK - Object Detection

This sample shows how to detect custom objects using the official Pytorch implementation of YOLOv8 from a ZED camera and ingest them into the ZED SDK to extract 3D informations and tracking for each objects.

Getting Started

Setting up

  • Install yolov8 using pip
pip install ultralytics

Run the program

NOTE: The ZED v1 is not compatible with this module

python detector.py --weights yolov8m.pt # [--img_size 512 --conf_thres 0.1 --svo path/to/file.svo]

Features

  • The camera point cloud is displayed in a 3D OpenGL view
  • 3D bounding boxes around detected objects are drawn
  • Objects classes and confidences can be changed

Training your own model

This sample can use any model trained with YOLOv8, including custom trained one. For a getting started on how to trained a model on a custom dataset with YOLOv5, see here https://docs.ultralytics.com/tutorials/train-custom-datasets/

Support

If you need assistance go to our Community site at https://community.stereolabs.com/