├── .gitignore ├── LICENSE ├── README.md ├── calib.py ├── data └── images │ └── 72b8c32d20b14b200833f9f32c612baa.jpg ├── engine.py ├── image_reader.py ├── model2trt.py ├── requirements.txt ├── resnet.py ├── trt_engine.py └── trt_inference.py /.gitignore: -------------------------------------------------------------------------------- 1 | *.pyc 2 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/musicbeer/pytorch-tensorrt/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/musicbeer/pytorch-tensorrt/HEAD/README.md -------------------------------------------------------------------------------- /calib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/musicbeer/pytorch-tensorrt/HEAD/calib.py -------------------------------------------------------------------------------- /data/images/72b8c32d20b14b200833f9f32c612baa.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/musicbeer/pytorch-tensorrt/HEAD/data/images/72b8c32d20b14b200833f9f32c612baa.jpg -------------------------------------------------------------------------------- /engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/musicbeer/pytorch-tensorrt/HEAD/engine.py -------------------------------------------------------------------------------- /image_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/musicbeer/pytorch-tensorrt/HEAD/image_reader.py -------------------------------------------------------------------------------- /model2trt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/musicbeer/pytorch-tensorrt/HEAD/model2trt.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/musicbeer/pytorch-tensorrt/HEAD/requirements.txt -------------------------------------------------------------------------------- /resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/musicbeer/pytorch-tensorrt/HEAD/resnet.py -------------------------------------------------------------------------------- /trt_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/musicbeer/pytorch-tensorrt/HEAD/trt_engine.py -------------------------------------------------------------------------------- /trt_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/musicbeer/pytorch-tensorrt/HEAD/trt_inference.py --------------------------------------------------------------------------------