├── .gitignore ├── CMakeLists.txt ├── LICENSE ├── README.md ├── data ├── .gitkeep └── README.md ├── docs ├── AddQBias-onnx.png ├── AddQBias.png ├── layernorm-onnx.png ├── layernorm.png ├── mask2pos.png ├── maskedSoftmax.png ├── mergeParams-onnx.png ├── mergeParams.png ├── naiveSlice-Tile-onnx.png ├── naiveSlice-Tile.png ├── onnx2trt_1.png ├── onnx2trt_1_2.png ├── onnx2trt_2.png ├── opt-AddQBias.png ├── opt-layernorm.png ├── opt-mask2pos.png ├── opt-maskedSoftmax.png ├── opt-mergeParams.png ├── opt-naiveSlice-Tile.png ├── pipeline.png ├── pytorch2onnx_1.png ├── pytorch2onnx_2.png └── pytorch2onnx_sol_2.png ├── export_dynamic_shape_onnx.sh ├── export_static_shape_onnx.sh ├── include ├── common.cuh ├── plugins │ ├── AddQBiasTransposePlugin.h │ ├── AddVBiasTransposePlugin.h │ ├── LayerNormPlugin.h │ ├── Mask2PosPlugin.h │ └── MaskedSoftmaxPlugin.h ├── reduce_kernel_utils.cuh └── tensorrt │ └── .gitkeep ├── inference_dynamic_shape_onnx.sh ├── inference_dynamic_shape_trt.sh ├── inference_static_shape_onnx.sh ├── model ├── .gitkeep ├── README.md ├── onnx │ └── .gitkeep └── plan │ └── .gitkeep ├── requirements.txt ├── so ├── plugins │ └── .gitkeep └── tensorrt │ └── .gitkeep ├── src ├── AnchorDETR │ ├── .gitignore │ ├── LICENSE │ ├── README.md │ ├── compute_flops.py │ ├── compute_speeds.py │ ├── datasets │ │ ├── __init__.py │ │ ├── coco.py │ │ ├── coco_eval.py │ │ ├── coco_panoptic.py │ │ ├── data_prefetcher.py │ │ ├── panoptic_eval.py │ │ ├── samplers.py │ │ ├── torchvision_datasets │ │ │ ├── __init__.py │ │ │ └── coco.py │ │ └── transforms.py │ ├── engine.py │ ├── export_onnx.py │ ├── inference_onnx.py │ ├── inference_trt.py │ ├── main.py │ ├── models │ │ ├── __init__.py │ │ ├── anchor_detr.py │ │ ├── backbone.py │ │ ├── matcher.py │ │ ├── row_column_decoupled_attention.py │ │ ├── segmentation.py │ │ └── transformer.py │ ├── requirements.txt │ └── util │ │ ├── __init__.py │ │ ├── box_ops.py │ │ ├── misc.py │ │ └── plot_utils.py ├── plugins │ ├── AddQBiasTransposePlugin.cu │ ├── AddVBiasTransposePlugin.cu │ ├── CMakeLists.txt │ ├── LayerNormPlugin.cu │ ├── Mask2PosPlugin.cu │ ├── MaskedSoftmaxPlugin.cu │ ├── testAddQBiasTransposePluginOnnx.py │ ├── testAddVBiasTransposePluginOnnx.py │ ├── testLayerNormPluginOnnx.py │ ├── testMask2PosPluginOnnx.py │ └── testMaskedSoftmaxPluginOnnx.py └── python │ ├── export_AnchorDETR.py │ └── modify_AnchorDETR.py └── test.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/.gitignore -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/README.md -------------------------------------------------------------------------------- /data/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/data/README.md -------------------------------------------------------------------------------- /docs/AddQBias-onnx.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/docs/AddQBias-onnx.png -------------------------------------------------------------------------------- /docs/AddQBias.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/docs/AddQBias.png -------------------------------------------------------------------------------- /docs/layernorm-onnx.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/docs/layernorm-onnx.png -------------------------------------------------------------------------------- /docs/layernorm.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/docs/layernorm.png -------------------------------------------------------------------------------- /docs/mask2pos.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/docs/mask2pos.png -------------------------------------------------------------------------------- /docs/maskedSoftmax.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/docs/maskedSoftmax.png -------------------------------------------------------------------------------- /docs/mergeParams-onnx.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/docs/mergeParams-onnx.png -------------------------------------------------------------------------------- /docs/mergeParams.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/docs/mergeParams.png -------------------------------------------------------------------------------- /docs/naiveSlice-Tile-onnx.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/docs/naiveSlice-Tile-onnx.png -------------------------------------------------------------------------------- /docs/naiveSlice-Tile.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/docs/naiveSlice-Tile.png -------------------------------------------------------------------------------- /docs/onnx2trt_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/docs/onnx2trt_1.png -------------------------------------------------------------------------------- /docs/onnx2trt_1_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/docs/onnx2trt_1_2.png -------------------------------------------------------------------------------- /docs/onnx2trt_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/docs/onnx2trt_2.png -------------------------------------------------------------------------------- /docs/opt-AddQBias.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/docs/opt-AddQBias.png -------------------------------------------------------------------------------- /docs/opt-layernorm.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/docs/opt-layernorm.png -------------------------------------------------------------------------------- /docs/opt-mask2pos.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/docs/opt-mask2pos.png -------------------------------------------------------------------------------- /docs/opt-maskedSoftmax.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/docs/opt-maskedSoftmax.png -------------------------------------------------------------------------------- /docs/opt-mergeParams.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/docs/opt-mergeParams.png -------------------------------------------------------------------------------- /docs/opt-naiveSlice-Tile.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/docs/opt-naiveSlice-Tile.png -------------------------------------------------------------------------------- /docs/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/docs/pipeline.png -------------------------------------------------------------------------------- /docs/pytorch2onnx_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/docs/pytorch2onnx_1.png -------------------------------------------------------------------------------- /docs/pytorch2onnx_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/docs/pytorch2onnx_2.png -------------------------------------------------------------------------------- /docs/pytorch2onnx_sol_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/docs/pytorch2onnx_sol_2.png -------------------------------------------------------------------------------- /export_dynamic_shape_onnx.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/export_dynamic_shape_onnx.sh -------------------------------------------------------------------------------- /export_static_shape_onnx.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/export_static_shape_onnx.sh -------------------------------------------------------------------------------- /include/common.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/include/common.cuh -------------------------------------------------------------------------------- /include/plugins/AddQBiasTransposePlugin.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/include/plugins/AddQBiasTransposePlugin.h -------------------------------------------------------------------------------- /include/plugins/AddVBiasTransposePlugin.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/include/plugins/AddVBiasTransposePlugin.h -------------------------------------------------------------------------------- /include/plugins/LayerNormPlugin.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/include/plugins/LayerNormPlugin.h -------------------------------------------------------------------------------- /include/plugins/Mask2PosPlugin.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/include/plugins/Mask2PosPlugin.h -------------------------------------------------------------------------------- /include/plugins/MaskedSoftmaxPlugin.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/include/plugins/MaskedSoftmaxPlugin.h -------------------------------------------------------------------------------- /include/reduce_kernel_utils.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/include/reduce_kernel_utils.cuh -------------------------------------------------------------------------------- /include/tensorrt/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /inference_dynamic_shape_onnx.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/inference_dynamic_shape_onnx.sh -------------------------------------------------------------------------------- /inference_dynamic_shape_trt.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/inference_dynamic_shape_trt.sh -------------------------------------------------------------------------------- /inference_static_shape_onnx.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/inference_static_shape_onnx.sh -------------------------------------------------------------------------------- /model/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/model/README.md -------------------------------------------------------------------------------- /model/onnx/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/plan/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/requirements.txt -------------------------------------------------------------------------------- /so/plugins/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /so/tensorrt/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/AnchorDETR/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/.gitignore -------------------------------------------------------------------------------- /src/AnchorDETR/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/LICENSE -------------------------------------------------------------------------------- /src/AnchorDETR/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/README.md -------------------------------------------------------------------------------- /src/AnchorDETR/compute_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/compute_flops.py -------------------------------------------------------------------------------- /src/AnchorDETR/compute_speeds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/compute_speeds.py -------------------------------------------------------------------------------- /src/AnchorDETR/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/datasets/__init__.py -------------------------------------------------------------------------------- /src/AnchorDETR/datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/datasets/coco.py -------------------------------------------------------------------------------- /src/AnchorDETR/datasets/coco_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/datasets/coco_eval.py -------------------------------------------------------------------------------- /src/AnchorDETR/datasets/coco_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/datasets/coco_panoptic.py -------------------------------------------------------------------------------- /src/AnchorDETR/datasets/data_prefetcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/datasets/data_prefetcher.py -------------------------------------------------------------------------------- /src/AnchorDETR/datasets/panoptic_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/datasets/panoptic_eval.py -------------------------------------------------------------------------------- /src/AnchorDETR/datasets/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/datasets/samplers.py -------------------------------------------------------------------------------- /src/AnchorDETR/datasets/torchvision_datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/datasets/torchvision_datasets/__init__.py -------------------------------------------------------------------------------- /src/AnchorDETR/datasets/torchvision_datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/datasets/torchvision_datasets/coco.py -------------------------------------------------------------------------------- /src/AnchorDETR/datasets/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/datasets/transforms.py -------------------------------------------------------------------------------- /src/AnchorDETR/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/engine.py -------------------------------------------------------------------------------- /src/AnchorDETR/export_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/export_onnx.py -------------------------------------------------------------------------------- /src/AnchorDETR/inference_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/inference_onnx.py -------------------------------------------------------------------------------- /src/AnchorDETR/inference_trt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/inference_trt.py -------------------------------------------------------------------------------- /src/AnchorDETR/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/main.py -------------------------------------------------------------------------------- /src/AnchorDETR/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/models/__init__.py -------------------------------------------------------------------------------- /src/AnchorDETR/models/anchor_detr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/models/anchor_detr.py -------------------------------------------------------------------------------- /src/AnchorDETR/models/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/models/backbone.py -------------------------------------------------------------------------------- /src/AnchorDETR/models/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/models/matcher.py -------------------------------------------------------------------------------- /src/AnchorDETR/models/row_column_decoupled_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/models/row_column_decoupled_attention.py -------------------------------------------------------------------------------- /src/AnchorDETR/models/segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/models/segmentation.py -------------------------------------------------------------------------------- /src/AnchorDETR/models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/models/transformer.py -------------------------------------------------------------------------------- /src/AnchorDETR/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/requirements.txt -------------------------------------------------------------------------------- /src/AnchorDETR/util/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/util/__init__.py -------------------------------------------------------------------------------- /src/AnchorDETR/util/box_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/util/box_ops.py -------------------------------------------------------------------------------- /src/AnchorDETR/util/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/util/misc.py -------------------------------------------------------------------------------- /src/AnchorDETR/util/plot_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/AnchorDETR/util/plot_utils.py -------------------------------------------------------------------------------- /src/plugins/AddQBiasTransposePlugin.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/plugins/AddQBiasTransposePlugin.cu -------------------------------------------------------------------------------- /src/plugins/AddVBiasTransposePlugin.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/plugins/AddVBiasTransposePlugin.cu -------------------------------------------------------------------------------- /src/plugins/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/plugins/CMakeLists.txt -------------------------------------------------------------------------------- /src/plugins/LayerNormPlugin.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/plugins/LayerNormPlugin.cu -------------------------------------------------------------------------------- /src/plugins/Mask2PosPlugin.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/plugins/Mask2PosPlugin.cu -------------------------------------------------------------------------------- /src/plugins/MaskedSoftmaxPlugin.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/plugins/MaskedSoftmaxPlugin.cu -------------------------------------------------------------------------------- /src/plugins/testAddQBiasTransposePluginOnnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/plugins/testAddQBiasTransposePluginOnnx.py -------------------------------------------------------------------------------- /src/plugins/testAddVBiasTransposePluginOnnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/plugins/testAddVBiasTransposePluginOnnx.py -------------------------------------------------------------------------------- /src/plugins/testLayerNormPluginOnnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/plugins/testLayerNormPluginOnnx.py -------------------------------------------------------------------------------- /src/plugins/testMask2PosPluginOnnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/plugins/testMask2PosPluginOnnx.py -------------------------------------------------------------------------------- /src/plugins/testMaskedSoftmaxPluginOnnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/plugins/testMaskedSoftmaxPluginOnnx.py -------------------------------------------------------------------------------- /src/python/export_AnchorDETR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/python/export_AnchorDETR.py -------------------------------------------------------------------------------- /src/python/modify_AnchorDETR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/src/python/modify_AnchorDETR.py -------------------------------------------------------------------------------- /test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YukSing12/anchordetr_tensorrt/HEAD/test.sh --------------------------------------------------------------------------------