YOLOv5 is working with USB webcam, but how can I use CSI camera?

bujna94 · September 1, 2021, 9:35pm

Hi, I’m running YOLOv5 on Jetson Nano. It’s working fine with USB webcam with command -

sudo python3 detect.py --weights runs/exp24/weights/best.pt --source 1

But how can I run it with CSI camera? This command below is working to show the camera, but somehow I need to pass it to the YOLOv5 detect script.

gst-launch-1.0 nvarguscamerasrc sensor_id=0 ! \
   'video/x-raw(memory:NVMM),width=3280, height=2464, framerate=21/1, format=NV12' ! \
   nvvidconv flip-method=0 ! 'video/x-raw,width=960, height=720' ! \
   nvvidconv ! nvegltransform ! nveglglessink -e

AastaLLL · September 2, 2021, 3:46am

Hi,

YOLOv5 uses OpenCV to get the input camera data.
Please modify the sample via OpenCV+GStreamer.

You can find an example below:

github.com

NVIDIA-AI-IOT/jetcam/blob/master/jetcam/csi_camera.py

from .camera import Camera
import atexit
import cv2
import numpy as np
import threading
import traitlets


class CSICamera(Camera):
    
    capture_device = traitlets.Integer(default_value=0)
    capture_fps = traitlets.Integer(default_value=30)
    capture_width = traitlets.Integer(default_value=640)
    capture_height = traitlets.Integer(default_value=480)
    
    def __init__(self, *args, **kwargs):
        super(CSICamera, self).__init__(*args, **kwargs)
        try:
            self.cap = cv2.VideoCapture(self._gst_str(), cv2.CAP_GSTREAMER)

This file has been truncated. show original

Thanks.

Topic		Replies	Views
Run YOLOv5 on JETSON NANO with CSI-camera Jetson Nano camera , yolo	4	2037	June 29, 2022
Yolov5 on jetson nano with csi-camera Jetson Nano camera , yolo	2	3864	October 15, 2021
Putting YoloV5 on Jetson Nano 2GB Jetson Nano camera , opencv , gstreamer , yolo	2	4195	March 8, 2022
How to use USB Webcam in Jetson Nano with Python and OpenCV? Jetson Nano camera	15	5801	October 15, 2021
Yolov5 detect.py camera issue Jetson Orin NX opencv , yolo	6	718	December 18, 2023
CSI Camera cannot be turned on after installing the YOLOv5 dependency package Jetson AGX Xavier camera , opencv , ubuntu , gstreamer , python	4	859	March 16, 2023
Cannot access webcam with opencv2 in python Jetson Orin Nano camera , opencv	8	1224	December 19, 2023
Using CSI Camera for detecting object with yolov5 Jetson Nano camera , yolo	4	919	April 5, 2023
How to operate CSI/MIPI camera with YOLOv5 on Jetson Nano Jetson Nano camera , yolo	2	2314	November 9, 2022
How to use multiple cameras on the Jetson NANO? Jetson Nano camera	4	2459	March 21, 2022

YOLOv5 is working with USB webcam, but how can I use CSI camera?

Related topics