hertasecurity / gpu-nms
This repository contains the CUDA implementation of the paper "Work-efficient Parallel Non-Maximum Suppression Kernels".
☆12Updated 4 years ago
Alternatives and similar repositories for gpu-nms:
Users that are interested in gpu-nms are comparing it to the libraries listed below
- CUDA implementation of NMS for PyTorch☆85Updated 5 years ago
- mmdetection in python训练, libtorch in c++部署☆12Updated 4 years ago
- Implement yolov5 with Tensorrt C++ api, and integrate batchedNMSPlugin. A Python wrapper is also provided.☆50Updated 3 years ago
- Parallel CUDA implementation of NON maximum Suppression☆79Updated 4 years ago
- Simple demo of tensorrt plugin☆42Updated 3 years ago
- centernet, mobilenetv2, centerface☆52Updated 5 years ago
- Based of paper "Quantization and Training of Neural Networks for Efficient Integer-Arithmetic-Only Inference"☆61Updated 4 years ago
- ☆25Updated 3 years ago
- Useful tensorrt plugin. For pytorch and mmdetection model conversion.☆161Updated 3 months ago
- TensorRT plugin forDCNv2 layer in ONNX model☆59Updated 4 years ago
- Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.☆57Updated 5 years ago
- ☆78Updated 4 years ago
- Useful plugins for Pytorch1.5+ and TensorRT7/8.☆37Updated 3 years ago
- centernet pytorch model to torch script model☆30Updated 4 years ago
- centerNet Caffe inference CPP☆69Updated 5 years ago
- convert torch module to tensorrt network or tvm function☆89Updated 5 years ago
- This repository has been moved. The new location is in https://github.com/TexasInstruments/edgeai-tensorlab☆71Updated 9 months ago
- useful cuda code .☆42Updated 2 years ago
- nanodet int8 量化,实测推理2ms一帧!☆37Updated 3 years ago
- SSD net Training on pytorch and Implementation on libtorch☆50Updated 5 years ago
- U版yolov5 2.0的tensorrt加速☆37Updated 4 years ago
- This repository provides a sample to run yolov3 on int8 mode in tensorRT☆26Updated 5 years ago
- A C++ inference SDK based on TensorRT☆70Updated 4 years ago
- ☆49Updated last year
- CenterNet's inference.(C++)/基于CenterNet的旋转目标检测C++版☆28Updated 4 years ago
- This repository has been moved. The new location is in https://github.com/TexasInstruments/edgeai-tensorlabUpdated last year
- A resnet18 version of CenterNet(objects as points)☆124Updated 3 years ago
- A simple demo to use TensorRT with pytorch☆28Updated 5 years ago
- YOLOV3-Tiny TensorRT6.0 13个类别☆32Updated 4 years ago
- Lightweight Networks such as MobileNet, ShuffleNet and ThunderNet implemented in Pytorch☆72Updated 4 years ago