/Semantic-Segmentation-CUDA

Using CityScapes to perform semantic segmentation in the AV space. Optimizing for FPS by leveraging PyTorch's C++ API (LibTorch) and NVIDIA's CV-CUDA package to run these models at the highest FPS possible!

Primary LanguagePython

Stargazers