traveller59 / torch2trt
convert torch module to tensorrt network or tvm function
☆89Updated 4 years ago
Related projects ⓘ
Alternatives and complementary repositories for torch2trt
- ☆138Updated 5 years ago
- CUDA implementation of NMS for PyTorch☆85Updated 4 years ago
- [ICCV 2019] Harmonious Bottleneck on Two Orthogonal Dimensions, surpassing MobileNetV2☆102Updated 4 years ago
- Partial Order Pruning: for Best Speed/Accuracy Trade-off in Neural Architecture Search☆150Updated 4 years ago
- Android demo for dabnn☆20Updated 5 years ago
- Parallel CUDA implementation of NON maximum Suppression☆79Updated 4 years ago
- Benchmark of TVM quantized model on CUDA☆112Updated 4 years ago
- Implementation of NetAdapt☆30Updated 4 years ago
- PSROIAlign with multi-batch training support in PyTorch☆23Updated 5 years ago
- Implementations of CVPR 2019 paper Distilling Object Detectors with Fine-grained Feature Imitation☆27Updated 2 years ago
- ☆37Updated 6 years ago
- Light-Head R-CNN in PyTorch☆93Updated 5 years ago
- TensorRT Net Wrapper☆95Updated 5 years ago
- ☆45Updated 5 years ago
- TensorRT prelu and slice☆39Updated 6 years ago
- TensorRT plugin forDCNv2 layer in ONNX model☆58Updated 4 years ago
- Model shared. Top1:67.408/Top5:87.258☆48Updated 6 years ago
- Rotate RoI Align and Rotate Position Sensitive RoI Align Operation in Caffe☆15Updated 5 years ago
- Caffe implementation of ICCV 2017 & TPAMI 2018 paper - ThiNet☆46Updated 6 years ago
- PeleeNet in PyTorch☆40Updated 6 years ago
- A Gluon implementation of Mnasnet☆59Updated 5 years ago
- Added quantization layer into caffe (support a coarse level fixed point simulation)☆22Updated 7 years ago
- Lightweight Networks such as MobileNet, ShuffleNet and ThunderNet implemented in Pytorch☆72Updated 4 years ago
- This is a CNN Analyzer tool, based on Netscope by dgschwend/netscope☆40Updated 6 years ago
- Experiments based on CenterNet (more backbones, TensorRT deployment and mask head)☆222Updated 4 years ago
- ☆44Updated 5 years ago
- Simulate quantization and quantization aware training for MXNet-Gluon models.☆46Updated 4 years ago