Skip to content

Latest commit

 

History

History
92 lines (68 loc) · 3.64 KB

README.md

File metadata and controls

92 lines (68 loc) · 3.64 KB
Raw

PhyCV-CUDA

Welcome to PhyCV-CUDA! This is the CUDA/C++ version of the PhyCV (Physics-inspired Computer Vision) library. It is specifically optimized for edge devices with NVIDIA GPUs. This repo is developed by Jalali-Lab @ UCLA. You can find the poster presented in NVIDIA GTC 2024 here.

Introduction

PhyCV CUDA holds the source files needed to run the C++/CUDA versions of the PhyCV algorithms. The PhyCV CUDA source code can be built and run on any machine with C++/CUDA support. Testing and benchmarks are done on NVIDIA's Jetson Nano.

Folder Structure

  • assets: sample input images/videos, sample results, documentations.
  • includes: head files
  • src: This folder contains the source code of PhyCV CUDA.
    • main.cpp: serves as the entry point for the application.
    • options.cpp: processes command-line options.
    • video.cpp: runs VEViD on input images, videos, and camera feeds.
    • detect_net.cpp: uses the Jetson Inference library to run object detection.
    • vevid.cu: the implementations of the vevid algorithms.
    • kernels.cu: the implementations of the CUDA kernels required to run VEViD.

Get Started

Platforms

  • This repo uses CUDA to leverage parallel processing on the GPU. Make sure you have a compatible GPU and that you have set up CUDA before the installation.

  • This repo requires the jetson-inference library for object detection. Clone the repository and build the project from source following the instructions here.

  • This repo is tested on NVIDIA Jetson Nano 4GB with the following versions:

    • Jetpack: 4.6.4
    • CUDA: 10.2.300
    • OpenCV: 4.1.1

Instructions

# 1. Download the repo from GitHub
git clone https://github.com/JalaliLabUCLA/phycv-cuda
# 2. cd into the repo
cd phycv-cuda
# 3. Compile
make

Now you should see the executable output vevid in the directory. We list typical usages of the repo below

Run VEViD on the video feed from the camera at default 480p resolution.

./vevid

Run VEViD on the video feed from the camera at 720p resolution. Use -r <width>,<height> to specify the processed frame size.

./vevid -r 1280,720

Run VEViD on a single image file. indicate the file location after -i

./vevid -i ./assets/input_images/dark_road.jpeg

Run VEViD on a single video file. indicate the file location after -v

./vevid -v ./assets/input_videos/video_campus.mp4

If you want to save the processed video, indicate saving location after -w. Note that saving the processed video may cause latency. So when the -w flag is turned on, the on-screen display is turned off.

./vevid -v ./assets/input_videos/video_campus.mp4 -w ./output/enhanced_campus.mp4

The default parameters of VEViD are defined in includes/options.hpp. You can change these parameters by using the -p and -r flags. Use -p <PARAM=val> where PARAM is one of the VEViD parameters S, T, b, G and val is a floating point number. Multiple parameters can be set at the same time, e.g:

Set S to 0.4 and b to 0.2 (parameters must be separated with commas not spaces).

./vevid -p S=0.4,b=0.2

See all the options from the command line:

./vevid -h

Other Usages:

  • To enable object detection, add -d flag to the command.

  • To display timing information, add -t flag to the command.

  • To enable VEViD-Lite, add -l flag to the command.