├── LICENSE ├── README.md ├── calib_train_image └── 此处存放INT8量化需要的训练图像约30660张.txt ├── checkpoint └── 该文件夹存放训练好的DETR pth模型文件.txt ├── cpp ├── CMakeLists.txt ├── model │ └── ONNX模型文件.txt ├── res.jpg └── src │ └── main.cc ├── detr_pth2onnx.py ├── generate_batch_plan.py ├── inference_detr_onnx.py ├── inference_detr_trt.py ├── model ├── README.md ├── __init__.py ├── backbone.py ├── box_ops.py ├── detr.py ├── hubconf.py ├── matcher.py ├── misc.py ├── position_encoding.py ├── segmentation.py └── transformer.py ├── performance_accuracy_detr.py ├── performance_time_detr.py ├── pic ├── average_diff_percentage.png ├── bug1.png ├── bug1_onnx.png ├── bug2.png ├── bug2_onnx.png ├── bug3.png ├── bug4.png ├── latency_vs_throughput.png ├── test_fp16.jpg ├── test_fp32.jpg └── time.png ├── requirements.txt ├── trt_int8_quant.py └── trt_util ├── __init__.py ├── calibrator.py ├── common.py ├── plot_box.py ├── process_img.py └── trt_lite.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/README.md -------------------------------------------------------------------------------- /calib_train_image/此处存放INT8量化需要的训练图像约30660张.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /checkpoint/该文件夹存放训练好的DETR pth模型文件.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cpp/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/cpp/CMakeLists.txt -------------------------------------------------------------------------------- /cpp/model/ONNX模型文件.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cpp/res.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/cpp/res.jpg -------------------------------------------------------------------------------- /cpp/src/main.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/cpp/src/main.cc -------------------------------------------------------------------------------- /detr_pth2onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/detr_pth2onnx.py -------------------------------------------------------------------------------- /generate_batch_plan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/generate_batch_plan.py -------------------------------------------------------------------------------- /inference_detr_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/inference_detr_onnx.py -------------------------------------------------------------------------------- /inference_detr_trt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/inference_detr_trt.py -------------------------------------------------------------------------------- /model/README.md: -------------------------------------------------------------------------------- 1 | 声明：该文件夹下的代码来源于FaceBook开源项目：https://github.com/facebookresearch/detr 2 | 3 | 4 | 5 | -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/model/backbone.py -------------------------------------------------------------------------------- /model/box_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/model/box_ops.py -------------------------------------------------------------------------------- /model/detr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/model/detr.py -------------------------------------------------------------------------------- /model/hubconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/model/hubconf.py -------------------------------------------------------------------------------- /model/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/model/matcher.py -------------------------------------------------------------------------------- /model/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/model/misc.py -------------------------------------------------------------------------------- /model/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/model/position_encoding.py -------------------------------------------------------------------------------- /model/segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/model/segmentation.py -------------------------------------------------------------------------------- /model/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/model/transformer.py -------------------------------------------------------------------------------- /performance_accuracy_detr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/performance_accuracy_detr.py -------------------------------------------------------------------------------- /performance_time_detr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/performance_time_detr.py -------------------------------------------------------------------------------- /pic/average_diff_percentage.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/pic/average_diff_percentage.png -------------------------------------------------------------------------------- /pic/bug1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/pic/bug1.png -------------------------------------------------------------------------------- /pic/bug1_onnx.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/pic/bug1_onnx.png -------------------------------------------------------------------------------- /pic/bug2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/pic/bug2.png -------------------------------------------------------------------------------- /pic/bug2_onnx.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/pic/bug2_onnx.png -------------------------------------------------------------------------------- /pic/bug3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/pic/bug3.png -------------------------------------------------------------------------------- /pic/bug4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/pic/bug4.png -------------------------------------------------------------------------------- /pic/latency_vs_throughput.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/pic/latency_vs_throughput.png -------------------------------------------------------------------------------- /pic/test_fp16.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/pic/test_fp16.jpg -------------------------------------------------------------------------------- /pic/test_fp32.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/pic/test_fp32.jpg -------------------------------------------------------------------------------- /pic/time.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/pic/time.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/requirements.txt -------------------------------------------------------------------------------- /trt_int8_quant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/trt_int8_quant.py -------------------------------------------------------------------------------- /trt_util/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /trt_util/calibrator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/trt_util/calibrator.py -------------------------------------------------------------------------------- /trt_util/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/trt_util/common.py -------------------------------------------------------------------------------- /trt_util/plot_box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/trt_util/plot_box.py -------------------------------------------------------------------------------- /trt_util/process_img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/trt_util/process_img.py -------------------------------------------------------------------------------- /trt_util/trt_lite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DataXujing/TensorRT-DETR/HEAD/trt_util/trt_lite.py --------------------------------------------------------------------------------