├── README.md ├── classification ├── data.txt ├── data │ ├── __init__.py │ └── dataloader.py ├── quantization.py ├── shot.jpg ├── tensorrt_PTA_classification_pipline.py ├── test_int8trt.py ├── test_torch.py ├── torch2onnx.py └── trt │ ├── __init__.py │ ├── calibrator.py │ ├── common.py │ └── utils.py └── detection ├── calibrator.py ├── coco.label ├── demo_onnx.py ├── demo_trt.py ├── dog.jpg ├── quantization.py ├── sample.py ├── show_img ├── output_onnx.jpg └── output_trt.jpg └── util_trt.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuanandsix/Tensorrt-int8-quantization-pipline/HEAD/README.md -------------------------------------------------------------------------------- /classification/data.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuanandsix/Tensorrt-int8-quantization-pipline/HEAD/classification/data.txt -------------------------------------------------------------------------------- /classification/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /classification/data/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuanandsix/Tensorrt-int8-quantization-pipline/HEAD/classification/data/dataloader.py -------------------------------------------------------------------------------- /classification/quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuanandsix/Tensorrt-int8-quantization-pipline/HEAD/classification/quantization.py -------------------------------------------------------------------------------- /classification/shot.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuanandsix/Tensorrt-int8-quantization-pipline/HEAD/classification/shot.jpg -------------------------------------------------------------------------------- /classification/tensorrt_PTA_classification_pipline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuanandsix/Tensorrt-int8-quantization-pipline/HEAD/classification/tensorrt_PTA_classification_pipline.py -------------------------------------------------------------------------------- /classification/test_int8trt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuanandsix/Tensorrt-int8-quantization-pipline/HEAD/classification/test_int8trt.py -------------------------------------------------------------------------------- /classification/test_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuanandsix/Tensorrt-int8-quantization-pipline/HEAD/classification/test_torch.py -------------------------------------------------------------------------------- /classification/torch2onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuanandsix/Tensorrt-int8-quantization-pipline/HEAD/classification/torch2onnx.py -------------------------------------------------------------------------------- /classification/trt/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /classification/trt/calibrator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuanandsix/Tensorrt-int8-quantization-pipline/HEAD/classification/trt/calibrator.py -------------------------------------------------------------------------------- /classification/trt/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuanandsix/Tensorrt-int8-quantization-pipline/HEAD/classification/trt/common.py -------------------------------------------------------------------------------- /classification/trt/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuanandsix/Tensorrt-int8-quantization-pipline/HEAD/classification/trt/utils.py -------------------------------------------------------------------------------- /detection/calibrator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuanandsix/Tensorrt-int8-quantization-pipline/HEAD/detection/calibrator.py -------------------------------------------------------------------------------- /detection/coco.label: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuanandsix/Tensorrt-int8-quantization-pipline/HEAD/detection/coco.label -------------------------------------------------------------------------------- /detection/demo_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuanandsix/Tensorrt-int8-quantization-pipline/HEAD/detection/demo_onnx.py -------------------------------------------------------------------------------- /detection/demo_trt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuanandsix/Tensorrt-int8-quantization-pipline/HEAD/detection/demo_trt.py -------------------------------------------------------------------------------- /detection/dog.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuanandsix/Tensorrt-int8-quantization-pipline/HEAD/detection/dog.jpg -------------------------------------------------------------------------------- /detection/quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuanandsix/Tensorrt-int8-quantization-pipline/HEAD/detection/quantization.py -------------------------------------------------------------------------------- /detection/sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuanandsix/Tensorrt-int8-quantization-pipline/HEAD/detection/sample.py -------------------------------------------------------------------------------- /detection/show_img/output_onnx.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuanandsix/Tensorrt-int8-quantization-pipline/HEAD/detection/show_img/output_onnx.jpg -------------------------------------------------------------------------------- /detection/show_img/output_trt.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuanandsix/Tensorrt-int8-quantization-pipline/HEAD/detection/show_img/output_trt.jpg -------------------------------------------------------------------------------- /detection/util_trt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuanandsix/Tensorrt-int8-quantization-pipline/HEAD/detection/util_trt.py --------------------------------------------------------------------------------