Real Time Object Detection using Jetson Nano by Ronailson Garcia de MoraisReal Time Object Detection using Jetson Nano by Ronailson Garcia de Morais

Real Time Object Detection using Jetson Nano

Ronailson Garcia de Morais

Ronailson Garcia de Morais

Original image: https://www.flickr.com/photos/nicolelee/19041780

Important

The input images are directly resized to match the input size of the model. I skipped adding the pad to the input image, it might affect the accuracy of the model if the input image has a different aspect ratio compared to the input size of the model. Always try to get an input size with a ratio close to the input images you will use.

Requirements

Check the requirements.txt file.
For ONNX, if you have a NVIDIA GPU, then install the onnxruntime-gpu, otherwise use the onnxruntime library.

Installation

git clone https://github.com/ibaiGorordo/ONNX-YOLOv8-Object-Detection.git cd ONNX-YOLOv8-Object-Detection pip install -r requirements.txt

ONNX Runtime

For Nvidia GPU computers: pip install onnxruntime-gpu
Otherwise: pip install onnxruntime

ONNX model

Use the Google Colab notebook to convert the model:
You can convert the model using the following code after installing ultralitics (pip install ultralytics):
from ultralytics import YOLO model = YOLO("yolov8m.pt") model.export(format="onnx", imgsz=[480,640])

Original YOLOv8 model

The original YOLOv8 model can be found in this repository: YOLOv8 Repository
The License of the models is GPL-3.0 license: License

Examples

Image inference:
python image_object_detection.py
Webcam inference:
python webcam_object_detection.py
python video_object_detection.py

References:

PINTO0309's model conversion tool: https://github.com/PINTO0309/openvino2tensorflow
Like this project

Posted Aug 31, 2024

Developed an efficient object detection system using NVIDIA Jetson, leveraging AI for real-time recognition and analysis in edge environments.