torchtrtz

Pytorch models built and saved to generate tensorrt weights file. Sibling project to use the generated trt weights file and …

Stars

2

Forks

3

Language

Python

Last Updated

Jun 22, 2021

Similar Repos
Repo Language Stars Description Updated At
Python 2 Converting Pytorch weights to ONNX & TensorRT engines May 18, 2022
Python 129 TensorFlow object detection models accelerated with NVIDIA TensorRT (TF-TRT) Apr 25, 2023
Python 119 PyTorch Lightning Optical Flow models, scripts, and pretrained weights. May 06, 2023
Python 60 Convert YOLOv3 and YOLOv3-tiny (PyTorch version) into TensorRT models. Apr 12, 2023
None 2 Directly exploring pytorch model convert to TensorRT engine. Jun 30, 2022
Jupyter Notebook 254 TensorFlow port of PyTorch Image Models (timm) - image models with pretrained weights. May 11, 2023
Jupyter Notebook 494 Human segmentation models, training/inference code, and trained weights, implemented in PyTorch Apr 29, 2023
Python 8 The original weights of some Caffe models, ported to PyTorch. Aug 11, 2022
Python 9 Pytorch refinenet for segmentation and pytorch -> onnx -> tensorrt Feb 11, 2022
Jupyter Notebook 7 Comparing Deep Learning Inference of Pytorch models running on CPU, CUDA and TensorRT May 18, 2022
Python 6 convert between pytorch, caffe prototxt/weights and darknet cfg/weights Jun 15, 2022
Python 19 convert between pytorch, caffe prototxt/weights and darknet cfg/weights May 17, 2021
C++ 19 Pilgrim Project: torch2trt, quick convert your pytorch model to TensorRT engine. Jun 30, 2022
Python 4 Generate Tikz figures from pytorch models Mar 29, 2023
None 2 PyTorch ,ONNX and TensorRT implementation of YOLOv4 May 19, 2021
Python 5 PyTorch ,ONNX and TensorRT implementation of YOLOv4 Aug 30, 2022
Python 7 Fast YOLO inference by means of TensorRT and PyTorch Jun 18, 2020
Python 2 YOLOv5 in Pytorch and TensorRT with ROS system implementation Jul 15, 2022
C++ 123 Useful tensorrt plugin. For pytorch and mmdetection model conversion. Aug 01, 2022
Jupyter Notebook 9 Notebooks and saved models for the AI4NP School Nov 23, 2021
Python 4 Generate an image file that contains a randomly generated maze Jan 29, 2023
Python 7129 Keras code and weights files for popular deep learning models. Aug 19, 2022
None 2 Keras code and weights files for popular deep learning models. Jan 13, 2022
Python 607 Music auto-tagging models and trained weights in keras/theano May 12, 2023
Python 2 A PyTorch implementation of BigGAN with pretrained weights and conversion scripts. Mar 09, 2022
Python 210 A tutorial about how to build a TensorRT Engine from a PyTorch Model with the … Jul 12, 2022
XSLT 2 Generated HTML5 Schema and Models for Tagsoup May 07, 2018
Python 947 🦋A PyTorch implementation of BigGAN with pretrained weights and conversion scripts. Jul 28, 2022
Python 31 Code and model definition for converting the official BYOL weights to PyTorch Mar 22, 2023
None 4 PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, … Mar 25, 2022
Python 5 PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, … Dec 13, 2021
Python 2 PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, … Aug 16, 2022
Python 3 PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, … Jun 22, 2022
None 3 PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, … Jul 24, 2022
Python 21395 PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, … Oct 17, 2022
Python 2 PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, … Feb 17, 2023
None 2 PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, … May 17, 2022
None 2 PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, … Apr 25, 2023
Python 11 PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, … Jun 07, 2023
None 2 PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, … Sep 18, 2023
C++ 7 Use win10 system and tensorrt to deploy the onnx model converted by pytorch Apr 23, 2022
Python 320 YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt … Aug 21, 2022
Python 19 YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt … May 20, 2022
Python 4 YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt … Feb 18, 2023
None 4 YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt … Sep 14, 2022
Python 2153 YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt … May 28, 2023
JavaScript 265 Creates ES6 ./index.js file in target directories that imports and exports all sibling files and … Aug 05, 2022
Python 105 Using VapourSynth with super resolution and interpolation models and speeding them up with TensorRT. Nov 29, 2022
Python 4 Using image super resolution models with vapoursynth and speeding them up with TensorRT. Aug 23, 2022
C 12 Simple code to read mangle masks, calculate ids and weights, and generate random points Apr 25, 2023