tensorrt yolo系列转trt模型plugin

资源文件列表(大概)

文件名

大小

tensorrtx-master/

tensorrtx-master/.clang-format

2.54KB

tensorrtx-master/.github/

tensorrtx-master/.github/ISSUE_TEMPLATE/

tensorrtx-master/.github/ISSUE_TEMPLATE/tensorrtx-issue-template.md

461B

tensorrtx-master/.github/stale.yml

684B

tensorrtx-master/.github/workflows/

tensorrtx-master/.github/workflows/pre-commit.yml

453B

tensorrtx-master/.gitignore

40B

tensorrtx-master/.pre-commit-config.yaml

521B

tensorrtx-master/LICENSE

1.05KB

tensorrtx-master/README.md

13.25KB

tensorrtx-master/alexnet/

tensorrtx-master/alexnet/CMakeLists.txt

686B

tensorrtx-master/alexnet/README.md

896B

tensorrtx-master/alexnet/alex.cpp

10.5KB

tensorrtx-master/alexnet/alexnet.py

8.71KB

tensorrtx-master/alexnet/logging.h

16.16KB

tensorrtx-master/arcface/

tensorrtx-master/arcface/CMakeLists.txt

1.66KB

tensorrtx-master/arcface/README.md

3.03KB

tensorrtx-master/arcface/arcface-mobilefacenet.cpp

18.77KB

tensorrtx-master/arcface/arcface-r100.cpp

19.97KB

tensorrtx-master/arcface/arcface-r50.cpp

17.13KB

tensorrtx-master/arcface/gen_wts.py

1.38KB

tensorrtx-master/arcface/logging.h

16.28KB

tensorrtx-master/arcface/macros.h

210B

tensorrtx-master/arcface/prelu.cu

6.83KB

tensorrtx-master/arcface/prelu.h

3.86KB

tensorrtx-master/centernet/

tensorrtx-master/centernet/README.md

1.12KB

tensorrtx-master/centernet/centernet.py

15.65KB

tensorrtx-master/centernet/dcnv2Plugin/

tensorrtx-master/centernet/dcnv2Plugin/CMakeLists.txt

867B

tensorrtx-master/centernet/dcnv2Plugin/dcn_v2_im2col_cuda.cu

19.33KB

tensorrtx-master/centernet/dcnv2Plugin/dcn_v2_im2col_cuda.h

5.12KB

tensorrtx-master/centernet/dcnv2Plugin/dcnv2Plugin.cpp

14.4KB

tensorrtx-master/centernet/dcnv2Plugin/dcnv2Plugin.h

4.33KB

tensorrtx-master/centernet/sample/

tensorrtx-master/centernet/sample/common.py

9.41KB

tensorrtx-master/centernet/sample/test.py

4.55KB

tensorrtx-master/crnn/

tensorrtx-master/crnn/CMakeLists.txt

879B

tensorrtx-master/crnn/README.md

1005B

tensorrtx-master/crnn/crnn.cpp

17.99KB

tensorrtx-master/crnn/genwts.py

859B

tensorrtx-master/crnn/logging.h

16.16KB

tensorrtx-master/csrnet/

tensorrtx-master/csrnet/CMakeLists.txt

667B

tensorrtx-master/csrnet/README.md

1.42KB

tensorrtx-master/csrnet/config.h

518B

tensorrtx-master/csrnet/csrnet.cpp

19.25KB

tensorrtx-master/csrnet/gen_wts.py

845B

tensorrtx-master/csrnet/logging.h

15.75KB

tensorrtx-master/csrnet/macros.h

211B

tensorrtx-master/dbnet/

tensorrtx-master/dbnet/CMakeLists.txt

850B

tensorrtx-master/dbnet/README.md

1.63KB

tensorrtx-master/dbnet/clipper/

tensorrtx-master/dbnet/clipper/CMakeLists.txt

118B

tensorrtx-master/dbnet/clipper/clipper.cpp

134.39KB

tensorrtx-master/dbnet/clipper/clipper.hpp

14.67KB

tensorrtx-master/dbnet/common.hpp

6.31KB

tensorrtx-master/dbnet/dbnet.cpp

23.1KB

tensorrtx-master/dbnet/logging.h

16.16KB

tensorrtx-master/dbnet/utils.h

3.07KB

tensorrtx-master/densenet/

tensorrtx-master/densenet/CMakeLists.txt

901B

tensorrtx-master/densenet/README.md

1.01KB

tensorrtx-master/densenet/densenet121.cpp

14.16KB

tensorrtx-master/densenet/densenet121.py

9.96KB

tensorrtx-master/densenet/logging.h

16.26KB

tensorrtx-master/detr/

tensorrtx-master/detr/CMakeLists.txt

929B

tensorrtx-master/detr/README.md

2.72KB

tensorrtx-master/detr/backbone.hpp

9.48KB

tensorrtx-master/detr/calibrator.hpp

4.06KB

tensorrtx-master/detr/common.hpp

2.66KB

tensorrtx-master/detr/detr.cpp

27.36KB

tensorrtx-master/detr/gen_wts.py

3.84KB

tensorrtx-master/detr/logging.h

16KB

tensorrtx-master/detr/macros.h

485B

tensorrtx-master/docker/

tensorrtx-master/docker/trt7/

tensorrtx-master/docker/trt7/Dockerfile

488B

tensorrtx-master/docker/trt8/

tensorrtx-master/docker/trt8/Dockerfile

82B

tensorrtx-master/docker/trt8/README.md

1.38KB

tensorrtx-master/efficient_ad/

tensorrtx-master/efficient_ad/CMakeLists.txt

1.37KB

tensorrtx-master/efficient_ad/README.md

1.33KB

tensorrtx-master/efficient_ad/datas/

tensorrtx-master/efficient_ad/datas/models/

tensorrtx-master/efficient_ad/datas/models/gen_wts.py

604B

tensorrtx-master/efficient_ad/efficientAD_det.cpp

9.76KB

tensorrtx-master/efficient_ad/src/

tensorrtx-master/efficient_ad/src/config.h

1.16KB

tensorrtx-master/efficient_ad/src/cuda_utils.h

843B

tensorrtx-master/efficient_ad/src/logging.h

16.31KB

tensorrtx-master/efficient_ad/src/macros.h

484B

tensorrtx-master/efficient_ad/src/model.cpp

18.7KB

tensorrtx-master/efficient_ad/src/model.h

358B

tensorrtx-master/efficient_ad/src/postprocess.h

280B

tensorrtx-master/efficient_ad/src/utils.h

812B

tensorrtx-master/efficientnet/

tensorrtx-master/efficientnet/CMakeLists.txt

744B

tensorrtx-master/efficientnet/README.md

916B

tensorrtx-master/efficientnet/efficientnet.cpp

9.8KB

tensorrtx-master/efficientnet/gen_wts.py

471B

tensorrtx-master/efficientnet/logging.h

16.16KB

tensorrtx-master/efficientnet/utils.hpp

9.22KB

tensorrtx-master/googlenet/

tensorrtx-master/googlenet/CMakeLists.txt

699B

tensorrtx-master/googlenet/README.md

943B

tensorrtx-master/googlenet/googlenet.cpp

13.93KB

tensorrtx-master/googlenet/logging.h

16.16KB

tensorrtx-master/hrnet/

tensorrtx-master/hrnet/hrnet-image-classification/

tensorrtx-master/hrnet/hrnet-image-classification/CMakeLists.txt

562B

tensorrtx-master/hrnet/hrnet-image-classification/README.md

1.18KB

tensorrtx-master/hrnet/hrnet-image-classification/common.hpp

12.73KB

tensorrtx-master/hrnet/hrnet-image-classification/demo.py

4.27KB

tensorrtx-master/hrnet/hrnet-image-classification/hrnet.cpp

47.51KB

tensorrtx-master/hrnet/hrnet-image-classification/logging.h

16.16KB

tensorrtx-master/hrnet/hrnet-semantic-segmentation/

tensorrtx-master/hrnet/hrnet-semantic-segmentation/CMakeLists.txt

1.1KB

tensorrtx-master/hrnet/hrnet-semantic-segmentation/README.md

3.19KB

tensorrtx-master/hrnet/hrnet-semantic-segmentation/common.hpp

15.57KB

tensorrtx-master/hrnet/hrnet-semantic-segmentation/gen_wts.py

1.7KB

tensorrtx-master/hrnet/hrnet-semantic-segmentation/hrnet.cpp

40.73KB

tensorrtx-master/hrnet/hrnet-semantic-segmentation/hrnet_ocr.cpp

46.42KB

tensorrtx-master/hrnet/hrnet-semantic-segmentation/hrnet_trt.py

7.53KB

tensorrtx-master/hrnet/hrnet-semantic-segmentation/logging.h

16.16KB

tensorrtx-master/ibnnet/

tensorrtx-master/ibnnet/CMakeLists.txt

964B

tensorrtx-master/ibnnet/InferenceEngine.cpp

3.63KB

tensorrtx-master/ibnnet/InferenceEngine.h

2.27KB

tensorrtx-master/ibnnet/README.md

998B

tensorrtx-master/ibnnet/gen_wts.py

837B

tensorrtx-master/ibnnet/holder.h

1.04KB

tensorrtx-master/ibnnet/ibnnet.cpp

8.23KB

tensorrtx-master/ibnnet/ibnnet.h

1.09KB

tensorrtx-master/ibnnet/layers.cpp

8.89KB

tensorrtx-master/ibnnet/layers.h

1.25KB

tensorrtx-master/ibnnet/logging.h

16.16KB

tensorrtx-master/ibnnet/main.cpp

2.87KB

tensorrtx-master/ibnnet/utils.cpp

1.13KB

tensorrtx-master/ibnnet/utils.h

897B

tensorrtx-master/inception/

tensorrtx-master/inception/inceptionv3/

tensorrtx-master/inception/inceptionv3/CMakeLists.txt

702B

tensorrtx-master/inception/inceptionv3/README.md

1.1KB

tensorrtx-master/inception/inceptionv3/inception_v3.cpp

19.96KB

tensorrtx-master/inception/inceptionv3/logging.h

16.16KB

tensorrtx-master/inception/inceptionv4/

tensorrtx-master/inception/inceptionv4/CMakeLists.txt

989B

tensorrtx-master/inception/inceptionv4/README.md

1.22KB

tensorrtx-master/inception/inceptionv4/inception_v4.cpp

8.75KB

tensorrtx-master/inception/inceptionv4/inception_v4.h

1.85KB

tensorrtx-master/inception/inceptionv4/layers_api.cpp

14.58KB

tensorrtx-master/inception/inceptionv4/layers_api.h

2.19KB

tensorrtx-master/inception/inceptionv4/logging.h

16.26KB

tensorrtx-master/inception/inceptionv4/main.cpp

2.26KB

tensorrtx-master/inception/inceptionv4/utils.cpp

1.12KB

tensorrtx-master/inception/inceptionv4/utils.h

608B

tensorrtx-master/lenet/

tensorrtx-master/lenet/CMakeLists.txt

801B

tensorrtx-master/lenet/README.md

1.1KB

tensorrtx-master/lenet/lenet.cpp

9.64KB

tensorrtx-master/lenet/lenet.py

5.86KB

tensorrtx-master/lenet/logging.h

16.2KB

tensorrtx-master/lenet/macros.h

211B

tensorrtx-master/lprnet/

tensorrtx-master/lprnet/1.jpg

2.09KB

tensorrtx-master/lprnet/CMakeLists.txt

1018B

tensorrtx-master/lprnet/LPRnet.cpp

19.88KB

tensorrtx-master/lprnet/README.md

776B

tensorrtx-master/lprnet/genwts.py

1.35KB

tensorrtx-master/lprnet/logging.h

16.16KB

tensorrtx-master/mlp/

tensorrtx-master/mlp/CMakeLists.txt

681B

tensorrtx-master/mlp/README.md

1.46KB

tensorrtx-master/mlp/logging.h

16.17KB

tensorrtx-master/mlp/mlp.cpp

10.06KB

tensorrtx-master/mlp/mlp.py

7.62KB

tensorrtx-master/mlp/mlp.wts

50B

tensorrtx-master/mnasnet/

tensorrtx-master/mnasnet/CMakeLists.txt

689B

tensorrtx-master/mnasnet/README.md

874B

tensorrtx-master/mnasnet/logging.h

16.16KB

tensorrtx-master/mnasnet/mnasnet.cpp

13.94KB

tensorrtx-master/mobilenet/

tensorrtx-master/mobilenet/mobilenetv2/

tensorrtx-master/mobilenet/mobilenetv2/CMakeLists.txt

702B

tensorrtx-master/mobilenet/mobilenetv2/README.md

1.42KB

tensorrtx-master/mobilenet/mobilenetv2/logging.h

16.16KB

tensorrtx-master/mobilenet/mobilenetv2/mobilenet_v2.cpp

14.12KB

tensorrtx-master/mobilenet/mobilenetv2/mobilenet_v2.py

10.24KB

tensorrtx-master/mobilenet/mobilenetv3/

tensorrtx-master/mobilenet/mobilenetv3/CMakeLists.txt

740B

tensorrtx-master/mobilenet/mobilenetv3/README.md

1.19KB

tensorrtx-master/mobilenet/mobilenetv3/logging.h

16.16KB

tensorrtx-master/mobilenet/mobilenetv3/mobilenet_v3.cpp

21.47KB

tensorrtx-master/mobilenet/mobilenetv3/mobilenet_v3.py

17.04KB

tensorrtx-master/psenet/

tensorrtx-master/psenet/CMakeLists.txt

958B

tensorrtx-master/psenet/README.md

1.32KB

tensorrtx-master/psenet/gen_tf_wts.py

851B

tensorrtx-master/psenet/layers.cpp

4.58KB

tensorrtx-master/psenet/layers.h

691B

tensorrtx-master/psenet/main.cpp

952B

tensorrtx-master/psenet/psenet.cpp

18.76KB

tensorrtx-master/psenet/psenet.h

1.23KB

tensorrtx-master/psenet/test.jpg

42.07KB

tensorrtx-master/psenet/utils.cpp

2.25KB

tensorrtx-master/psenet/utils.h

2.28KB

tensorrtx-master/rcnn/

tensorrtx-master/rcnn/BatchedNms.cu

6.62KB

tensorrtx-master/rcnn/BatchedNmsPlugin.h

6.53KB

tensorrtx-master/rcnn/CMakeLists.txt

1.3KB

tensorrtx-master/rcnn/MaskRcnnInference.cu

2.34KB

tensorrtx-master/rcnn/MaskRcnnInferencePlugin.h

6.01KB

tensorrtx-master/rcnn/PredictorDecode.cu

4.65KB

tensorrtx-master/rcnn/PredictorDecodePlugin.h

7.5KB

tensorrtx-master/rcnn/README.md

10.8KB

tensorrtx-master/rcnn/RoiAlign.cu

5.93KB

tensorrtx-master/rcnn/RoiAlignPlugin.h

6.9KB

tensorrtx-master/rcnn/RpnDecode.cu

5.7KB

tensorrtx-master/rcnn/RpnDecodePlugin.h

7.26KB

tensorrtx-master/rcnn/RpnNms.cu

5.1KB

tensorrtx-master/rcnn/RpnNmsPlugin.h

6.02KB

tensorrtx-master/rcnn/backbone.hpp

7.32KB

tensorrtx-master/rcnn/calibrator.hpp

4.08KB

tensorrtx-master/rcnn/common.hpp

2.99KB

tensorrtx-master/rcnn/cuda_utils.h

1.09KB

tensorrtx-master/rcnn/gen_wts.py

2.96KB

tensorrtx-master/rcnn/logging.h

16KB

tensorrtx-master/rcnn/macros.h

252B

tensorrtx-master/rcnn/rcnn.cpp

23.5KB

tensorrtx-master/real-esrgan/

tensorrtx-master/real-esrgan/CMakeLists.txt

1.18KB

tensorrtx-master/real-esrgan/README.md

2.02KB

tensorrtx-master/real-esrgan/common.hpp

5.66KB

tensorrtx-master/real-esrgan/cuda_utils.h

494B

tensorrtx-master/real-esrgan/gen_wts.py

4.15KB

tensorrtx-master/real-esrgan/logging.h

16.2KB

tensorrtx-master/real-esrgan/macros.h

462B

tensorrtx-master/real-esrgan/postprocess.cu

1.61KB

tensorrtx-master/real-esrgan/postprocess.hpp

5.58KB

tensorrtx-master/real-esrgan/preprocess.cu

1.56KB

tensorrtx-master/real-esrgan/preprocess.hpp

5.56KB

tensorrtx-master/real-esrgan/real-esrgan.cpp

12.01KB

tensorrtx-master/real-esrgan/utils.h

830B

tensorrtx-master/refinedet/

tensorrtx-master/refinedet/CMakeLists.txt

1.51KB

tensorrtx-master/refinedet/README.md

956B

tensorrtx-master/refinedet/calibrator.cpp

2.87KB

tensorrtx-master/refinedet/calibrator.h

1.15KB

tensorrtx-master/refinedet/configure.h

1.9KB

tensorrtx-master/refinedet/gen_wts_refinedet.py

709B

tensorrtx-master/refinedet/logging.h

16.16KB

tensorrtx-master/refinedet/refinedet.cpp

42.88KB

tensorrtx-master/refinedet/utils.h

1.86KB

tensorrtx-master/repvgg/

tensorrtx-master/repvgg/CMakeLists.txt

685B

tensorrtx-master/repvgg/README.md

985B

tensorrtx-master/repvgg/gen_wts.py

895B

tensorrtx-master/repvgg/logging.h

1.13KB

tensorrtx-master/repvgg/repvgg.cpp

12.63KB

tensorrtx-master/resnet/

tensorrtx-master/resnet/CMakeLists.txt

1.25KB

tensorrtx-master/resnet/README.md

2.79KB

tensorrtx-master/resnet/logging.h

16.16KB

tensorrtx-master/resnet/resnet18.cpp

12.57KB

tensorrtx-master/resnet/resnet34.cpp

13.44KB

tensorrtx-master/resnet/resnet50.cpp

13.54KB

tensorrtx-master/resnet/resnet50.py

9.97KB

tensorrtx-master/resnet/resnext50_32x4d.cpp

13.56KB

tensorrtx-master/resnet/wide_resnet50.py

9.5KB

tensorrtx-master/resnet/wideresnet50.cpp

13.55KB

tensorrtx-master/retinaface/

tensorrtx-master/retinaface/CMakeLists.txt

1.44KB

tensorrtx-master/retinaface/README.md

2.62KB

tensorrtx-master/retinaface/calibrator.cpp

2.78KB

tensorrtx-master/retinaface/calibrator.h

1.22KB

tensorrtx-master/retinaface/common.hpp

6.95KB

tensorrtx-master/retinaface/decode.cu

8.88KB

tensorrtx-master/retinaface/decode.h

4.08KB

tensorrtx-master/retinaface/logging.h

16.2KB

tensorrtx-master/retinaface/macros.h

211B

tensorrtx-master/retinaface/retina_mnet.cpp

18.5KB

tensorrtx-master/retinaface/retina_r50.cpp

19.41KB

tensorrtx-master/retinaface/retinaface_trt.py

11.85KB

tensorrtx-master/retinafaceAntiCov/

tensorrtx-master/retinafaceAntiCov/CMakeLists.txt

1.43KB

tensorrtx-master/retinafaceAntiCov/README.md

1.55KB

tensorrtx-master/retinafaceAntiCov/decode.cu

7.68KB

tensorrtx-master/retinafaceAntiCov/decode.h

4.31KB

tensorrtx-master/retinafaceAntiCov/gen_wts.py

736B

tensorrtx-master/retinafaceAntiCov/logging.h

16.19KB

tensorrtx-master/retinafaceAntiCov/macros.h

210B

tensorrtx-master/retinafaceAntiCov/retinafaceAntiCov.cpp

26.63KB

tensorrtx-master/scaled-yolov4/

tensorrtx-master/scaled-yolov4/CMakeLists.txt

1.11KB

tensorrtx-master/scaled-yolov4/README.md

2.07KB

tensorrtx-master/scaled-yolov4/common.hpp

6.89KB

tensorrtx-master/scaled-yolov4/gen_wts.py

731B

tensorrtx-master/scaled-yolov4/logging.h

16.26KB

tensorrtx-master/scaled-yolov4/mish.cu

5.62KB

tensorrtx-master/scaled-yolov4/mish.h

3.48KB

tensorrtx-master/scaled-yolov4/utils.h

1.15KB

tensorrtx-master/scaled-yolov4/yololayer.cu

9.25KB

tensorrtx-master/scaled-yolov4/yololayer.h

4.55KB

tensorrtx-master/scaled-yolov4/yolov4_csp.cpp

26.72KB

tensorrtx-master/senet/

tensorrtx-master/senet/CMakeLists.txt

697B

tensorrtx-master/senet/README.md

878B

tensorrtx-master/senet/logging.h

16.16KB

tensorrtx-master/senet/se_resnet50.cpp

14.39KB

tensorrtx-master/shufflenetv2/

tensorrtx-master/shufflenetv2/CMakeLists.txt

707B

tensorrtx-master/shufflenetv2/README.md

1.21KB

tensorrtx-master/shufflenetv2/logging.h

16.16KB

tensorrtx-master/shufflenetv2/shufflenet_v2.cpp

15.5KB

tensorrtx-master/squeezenet/

tensorrtx-master/squeezenet/CMakeLists.txt

704B

tensorrtx-master/squeezenet/README.md

880B

tensorrtx-master/squeezenet/logging.h

16.16KB

tensorrtx-master/squeezenet/squeezenet.cpp

10.9KB

tensorrtx-master/superpoint/

tensorrtx-master/superpoint/CMakeLists.txt

989B

tensorrtx-master/superpoint/README.md

2.57KB

tensorrtx-master/superpoint/gen_wts.py

505B

tensorrtx-master/superpoint/logging.h

16.57KB

tensorrtx-master/superpoint/supernet.cpp

7.95KB

tensorrtx-master/superpoint/utils.cpp

2.52KB

tensorrtx-master/superpoint/utils.h

982B

tensorrtx-master/swin-transformer/

tensorrtx-master/swin-transformer/semantic-segmentation/

tensorrtx-master/swin-transformer/semantic-segmentation/CMakeLists.txt

3.06KB

tensorrtx-master/swin-transformer/semantic-segmentation/README.md

1.35KB

tensorrtx-master/swin-transformer/semantic-segmentation/UpsampleKernel.cu

4.89KB

tensorrtx-master/swin-transformer/semantic-segmentation/UpsamplePlugin.cpp

6.55KB

tensorrtx-master/swin-transformer/semantic-segmentation/UpsamplePlugin.h

2.46KB

tensorrtx-master/swin-transformer/semantic-segmentation/UpsmapleKernel.h

333B

tensorrtx-master/swin-transformer/semantic-segmentation/common.hpp

34.86KB

tensorrtx-master/swin-transformer/semantic-segmentation/fillmask.cu

5.13KB

tensorrtx-master/swin-transformer/semantic-segmentation/fillmask.h

3.47KB

tensorrtx-master/swin-transformer/semantic-segmentation/gelu.cu

4.99KB

tensorrtx-master/swin-transformer/semantic-segmentation/gelu.h

3.47KB

tensorrtx-master/swin-transformer/semantic-segmentation/gen_wts.py

592B

tensorrtx-master/swin-transformer/semantic-segmentation/include/

tensorrtx-master/swin-transformer/semantic-segmentation/include/dirent.h

27.31KB

tensorrtx-master/swin-transformer/semantic-segmentation/layerNorm.cu

5.14KB

tensorrtx-master/swin-transformer/semantic-segmentation/layerNorm.h

3.51KB

tensorrtx-master/swin-transformer/semantic-segmentation/logging.h

16.65KB

tensorrtx-master/swin-transformer/semantic-segmentation/main.cpp

57B

tensorrtx-master/swin-transformer/semantic-segmentation/myhpp.h

754B

tensorrtx-master/swin-transformer/semantic-segmentation/trainsform.cpp

14.21KB

tensorrtx-master/swin-transformer/semantic-segmentation/utilsn.h

3.01KB

tensorrtx-master/tsm/

tensorrtx-master/tsm/CMakeLists.txt

685B

tensorrtx-master/tsm/README.md

3.17KB

tensorrtx-master/tsm/demo.sh

2.38KB

tensorrtx-master/tsm/gen_wts.py

1.54KB

tensorrtx-master/tsm/logging.h

16.16KB

tensorrtx-master/tsm/mmaction2_tsm_r50_config.py

563B

tensorrtx-master/tsm/test_shift.py

7.53KB

tensorrtx-master/tsm/tsm_r50.cpp

16.75KB

tensorrtx-master/tsm/tsm_r50.py

18.8KB

tensorrtx-master/tutorials/

tensorrtx-master/tutorials/check_fp16_int8_support.md

506B

tensorrtx-master/tutorials/contribution.md

731B

tensorrtx-master/tutorials/faq.md

3.54KB

tensorrtx-master/tutorials/from_pytorch_to_trt_stepbystep_hrnet.md

5.48KB

tensorrtx-master/tutorials/getting_started.md

3.15KB

tensorrtx-master/tutorials/install.md

1.69KB

tensorrtx-master/tutorials/measure_performance.md

3.54KB

tensorrtx-master/tutorials/migrating_from_tensorrt_4_to_7.md

635B

tensorrtx-master/tutorials/multi_GPU_processing.md

1.97KB

tensorrtx-master/tutorials/run_on_windows.md

4.93KB

tensorrtx-master/ufld/

tensorrtx-master/ufld/CMakeLists.txt

695B

tensorrtx-master/ufld/README.md

1.36KB

tensorrtx-master/ufld/common.hpp

6.28KB

tensorrtx-master/ufld/gen_wts.py

613B

tensorrtx-master/ufld/lane_det.cpp

15.74KB

tensorrtx-master/ufld/logging.h

15.58KB

tensorrtx-master/ufld/macros.h

462B

tensorrtx-master/ufld/pth2onnx.py

579B

tensorrtx-master/unet/

tensorrtx-master/unet/CMakeLists.txt

718B

tensorrtx-master/unet/README.md

1.9KB

tensorrtx-master/unet/common.hpp

2.88KB

tensorrtx-master/unet/gen_wts.py

538B

tensorrtx-master/unet/logging.h

16.2KB

tensorrtx-master/unet/macros.h

462B

tensorrtx-master/unet/unet.cpp

12KB

tensorrtx-master/vgg/

tensorrtx-master/vgg/CMakeLists.txt

671B

tensorrtx-master/vgg/README.md

761B

tensorrtx-master/vgg/logging.h

16.16KB

tensorrtx-master/vgg/vgg11.cpp

10.9KB

tensorrtx-master/yolop/

tensorrtx-master/yolop/CMakeLists.txt

992B

tensorrtx-master/yolop/README.md

3.43KB

tensorrtx-master/yolop/common.hpp

14.89KB

tensorrtx-master/yolop/cuda_utils.h

344B

tensorrtx-master/yolop/gen_wts.py

1KB

tensorrtx-master/yolop/logging.h

606B

tensorrtx-master/yolop/macros.h

210B

tensorrtx-master/yolop/utils.h

1.49KB

tensorrtx-master/yolop/yololayer.cu

12.21KB

tensorrtx-master/yolop/yololayer.h

4.63KB

tensorrtx-master/yolop/yolop.cpp

8.21KB

tensorrtx-master/yolop/yolop.hpp

12.76KB

tensorrtx-master/yolop/yolop_trt.py

15.66KB

tensorrtx-master/yolov3-spp/

tensorrtx-master/yolov3-spp/CMakeLists.txt

1.09KB

tensorrtx-master/yolov3-spp/README.md

2.29KB

tensorrtx-master/yolov3-spp/Utils.h

3.07KB

tensorrtx-master/yolov3-spp/gen_wts.py

637B

tensorrtx-master/yolov3-spp/logging.h

16.16KB

tensorrtx-master/yolov3-spp/samples/

tensorrtx-master/yolov3-spp/samples/bus.jpg

476.01KB

tensorrtx-master/yolov3-spp/samples/zidane.jpg

164.99KB

tensorrtx-master/yolov3-spp/yololayer.cu

9.25KB

tensorrtx-master/yolov3-spp/yololayer.h

4.9KB

tensorrtx-master/yolov3-spp/yolov3-spp.cpp

27.04KB

tensorrtx-master/yolov3-tiny/

tensorrtx-master/yolov3-tiny/CMakeLists.txt

1.22KB

tensorrtx-master/yolov3-tiny/README.md

1.62KB

tensorrtx-master/yolov3-tiny/gen_wts.py

750B

tensorrtx-master/yolov3-tiny/logging.h

16.19KB

tensorrtx-master/yolov3-tiny/macros.h

210B

tensorrtx-master/yolov3-tiny/utils.h

3.12KB

tensorrtx-master/yolov3-tiny/yololayer.cu

8.92KB

tensorrtx-master/yolov3-tiny/yololayer.h

4.79KB

tensorrtx-master/yolov3-tiny/yolov3-tiny.cpp

18.23KB

tensorrtx-master/yolov3/

tensorrtx-master/yolov3/CMakeLists.txt

1.17KB

tensorrtx-master/yolov3/README.md

2.57KB

tensorrtx-master/yolov3/calibrator.cpp

2.79KB

tensorrtx-master/yolov3/calibrator.h

1.22KB

tensorrtx-master/yolov3/gen_wts.py

740B

tensorrtx-master/yolov3/logging.h

16.28KB

tensorrtx-master/yolov3/macros.h

211B

tensorrtx-master/yolov3/utils.h

2.49KB

tensorrtx-master/yolov3/yololayer.cu

8.96KB

tensorrtx-master/yolov3/yololayer.h

4.92KB

tensorrtx-master/yolov3/yolov3.cpp

24.7KB

tensorrtx-master/yolov3/yolov3_trt.py

18.12KB

tensorrtx-master/yolov4/

tensorrtx-master/yolov4/CMakeLists.txt

1.1KB

tensorrtx-master/yolov4/README.md

1.97KB

tensorrtx-master/yolov4/gen_wts.py

719B

tensorrtx-master/yolov4/logging.h

16.16KB

tensorrtx-master/yolov4/mish.cu

5.62KB

tensorrtx-master/yolov4/mish.h

3.45KB

tensorrtx-master/yolov4/utils.h

1.16KB

tensorrtx-master/yolov4/yololayer.cu

9.19KB

tensorrtx-master/yolov4/yololayer.h

4.55KB

tensorrtx-master/yolov4/yolov4.cpp

32.57KB

tensorrtx-master/yolov5/

tensorrtx-master/yolov5/CMakeLists.txt

1.76KB

tensorrtx-master/yolov5/README.md

10.67KB

tensorrtx-master/yolov5/gen_wts.py

2.2KB

tensorrtx-master/yolov5/images

21B

tensorrtx-master/yolov5/plugin/

tensorrtx-master/yolov5/plugin/yololayer.cu

10.18KB

tensorrtx-master/yolov5/plugin/yololayer.h

3.52KB

tensorrtx-master/yolov5/src/

tensorrtx-master/yolov5/src/calibrator.cpp

3.29KB

tensorrtx-master/yolov5/src/calibrator.h

1.18KB

tensorrtx-master/yolov5/src/config.h

2.01KB

tensorrtx-master/yolov5/src/cuda_utils.h

417B

tensorrtx-master/yolov5/src/logging.h

16.2KB

tensorrtx-master/yolov5/src/macros.h

484B

tensorrtx-master/yolov5/src/model.cpp

33.3KB

tensorrtx-master/yolov5/src/model.h

1022B

tensorrtx-master/yolov5/src/postprocess.cpp

6.8KB

tensorrtx-master/yolov5/src/postprocess.h

711B

tensorrtx-master/yolov5/src/preprocess.cu

4.46KB

tensorrtx-master/yolov5/src/preprocess.h

522B

tensorrtx-master/yolov5/src/types.h

275B

tensorrtx-master/yolov5/src/utils.h

1.96KB

tensorrtx-master/yolov5/yolov5_cls.cpp

9.33KB

tensorrtx-master/yolov5/yolov5_cls_trt.py

8.69KB

tensorrtx-master/yolov5/yolov5_det.cpp

7.6KB

tensorrtx-master/yolov5/yolov5_det_cuda_python.py

17.96KB

tensorrtx-master/yolov5/yolov5_det_trt.py

17.87KB

tensorrtx-master/yolov5/yolov5_seg.cpp

8.59KB

tensorrtx-master/yolov5/yolov5_seg_trt.py

22.29KB

tensorrtx-master/yolov7/

tensorrtx-master/yolov7/CMakeLists.txt

1.36KB

tensorrtx-master/yolov7/README.md

3.01KB

tensorrtx-master/yolov7/gen_wts.py

1.72KB

tensorrtx-master/yolov7/images

21B

tensorrtx-master/yolov7/include/

tensorrtx-master/yolov7/include/block.h

1.5KB

tensorrtx-master/yolov7/include/calibrator.h

1.22KB

tensorrtx-master/yolov7/include/config.h

1.73KB

tensorrtx-master/yolov7/include/cuda_utils.h

417B

tensorrtx-master/yolov7/include/logging.h

16.2KB

tensorrtx-master/yolov7/include/macros.h

484B

tensorrtx-master/yolov7/include/model.h

1.33KB

tensorrtx-master/yolov7/include/postprocess.h

465B

tensorrtx-master/yolov7/include/preprocess.h

542B

tensorrtx-master/yolov7/include/types.h

257B

tensorrtx-master/yolov7/include/utils.h

1.43KB

tensorrtx-master/yolov7/main.cpp

7.6KB

tensorrtx-master/yolov7/plugin/

tensorrtx-master/yolov7/plugin/yololayer.cu

10.29KB

tensorrtx-master/yolov7/plugin/yololayer.h

3.5KB

tensorrtx-master/yolov7/src/

tensorrtx-master/yolov7/src/block.cpp

11.18KB

tensorrtx-master/yolov7/src/calibrator.cpp

2.78KB

tensorrtx-master/yolov7/src/model.cpp

148.18KB

tensorrtx-master/yolov7/src/postprocess.cpp

3.22KB

tensorrtx-master/yolov7/src/preprocess.cu

4.45KB

tensorrtx-master/yolov7/yolov7_trt.py

17.77KB

tensorrtx-master/yolov8/

tensorrtx-master/yolov8/CMakeLists.txt

1.97KB

tensorrtx-master/yolov8/README.md

6.8KB

tensorrtx-master/yolov8/gen_wts.py

1.76KB

tensorrtx-master/yolov8/include/

tensorrtx-master/yolov8/include/block.h

1.75KB

tensorrtx-master/yolov8/include/calibrator.h

1.21KB

tensorrtx-master/yolov8/include/config.h

928B

tensorrtx-master/yolov8/include/cuda_utils.h

417B

tensorrtx-master/yolov8/include/logging.h

16.2KB

tensorrtx-master/yolov8/include/macros.h

484B

tensorrtx-master/yolov8/include/model.h

2.01KB

tensorrtx-master/yolov8/include/postprocess.h

1.34KB

tensorrtx-master/yolov8/include/preprocess.h

440B

tensorrtx-master/yolov8/include/types.h

423B

tensorrtx-master/yolov8/include/utils.h

2.53KB

tensorrtx-master/yolov8/plugin/

tensorrtx-master/yolov8/plugin/yololayer.cu

12.4KB

tensorrtx-master/yolov8/plugin/yololayer.h

4.14KB

tensorrtx-master/yolov8/src/

tensorrtx-master/yolov8/src/block.cpp

12.83KB

tensorrtx-master/yolov8/src/calibrator.cpp

2.83KB

tensorrtx-master/yolov8/src/model.cpp

124.62KB

tensorrtx-master/yolov8/src/postprocess.cpp

10.56KB

tensorrtx-master/yolov8/src/postprocess.cu

2.97KB

tensorrtx-master/yolov8/src/preprocess.cu

4.76KB

tensorrtx-master/yolov8/yolov8_cls.cpp

9.82KB

tensorrtx-master/yolov8/yolov8_cls_trt.py

9.83KB

tensorrtx-master/yolov8/yolov8_det.cpp

10.91KB

tensorrtx-master/yolov8/yolov8_det_trt.py

18.04KB

tensorrtx-master/yolov8/yolov8_pose.cpp

10.92KB

tensorrtx-master/yolov8/yolov8_pose_trt.py

19.37KB

tensorrtx-master/yolov8/yolov8_seg.cpp

13.19KB

tensorrtx-master/yolov8/yolov8_seg_trt.py

22.46KB

tensorrtx-master/yolov9/

tensorrtx-master/yolov9/CMakeLists.txt

2.09KB

tensorrtx-master/yolov9/README.md

3.95KB

tensorrtx-master/yolov9/demo.cpp

9.04KB

tensorrtx-master/yolov9/gen_wts.py

2.63KB

tensorrtx-master/yolov9/images

21B

tensorrtx-master/yolov9/include/

tensorrtx-master/yolov9/include/block.h

3.35KB

tensorrtx-master/yolov9/include/calibrator.h

1.12KB

tensorrtx-master/yolov9/include/config.h

2.09KB

tensorrtx-master/yolov9/include/cuda_utils.h

417B

tensorrtx-master/yolov9/include/logging.h

16.2KB

tensorrtx-master/yolov9/include/macros.h

484B

tensorrtx-master/yolov9/include/model.h

2.52KB

tensorrtx-master/yolov9/include/postprocess.h

1.11KB

tensorrtx-master/yolov9/include/preprocess.h

522B

tensorrtx-master/yolov9/include/types.h

360B

tensorrtx-master/yolov9/include/utils.h

1.95KB

tensorrtx-master/yolov9/plugin/

tensorrtx-master/yolov9/plugin/yololayer.cu

8.31KB

tensorrtx-master/yolov9/plugin/yololayer.h

3.88KB

tensorrtx-master/yolov9/src/

tensorrtx-master/yolov9/src/block.cpp

22.66KB

tensorrtx-master/yolov9/src/calibrator.cpp

3.58KB

tensorrtx-master/yolov9/src/model.cpp

61.53KB

tensorrtx-master/yolov9/src/postprocess.cpp

9.43KB

tensorrtx-master/yolov9/src/postprocess.cu

2.99KB

tensorrtx-master/yolov9/src/preprocess.cu

4.75KB

tensorrtx-master/yolov9/windows/

tensorrtx-master/yolov9/windows/dirent.h

26.61KB

tensorrtx-master/yolov9/yolov9_trt.py

17.74KB

资源内容介绍

tensorrt yolo系列转trt模型plugin

# TensorRTxTensorRTx aims to implement popular deep learning networks with TensorRT network definition API.Why don't we use a parser (ONNX parser, UFF parser, caffe parser, etc), but use complex APIs to build a network from scratch? I have summarized the advantages in the following aspects.- **Flexible**, easy to modify the network, add/delete a layer or input/output tensor, replace a layer, merge layers, integrate preprocessing and postprocessing into network, etc.- **Debuggable**, construct the entire network in an incremental development manner, easy to get middle layer results.- **Educational**, learn about the network structure during this development, rather than treating everything as a black box.The basic workflow of TensorRTx is:1. Get the trained models from pytorch, mxnet or tensorflow, etc. Some pytorch models can be found in my repo [pytorchx](https://github.com/wang-xinyu/pytorchx), the remaining are from popular open-source repos.2. Export the weights to a plain text file -- [.wts file](./tutorials/getting_started.md#the-wts-content-format).3. Load weights in TensorRT, define the network, build a TensorRT engine.4. Load the TensorRT engine and run inference.## News- `21 Jun 2024`. [WuxinrongY](https://github.com/WuxinrongY): YOLOv9-T, YOLOv9-S, YOLOv9-M- `28 Apr 2024`. [lindsayshuo](https://github.com/lindsayshuo): YOLOv8-pose- `22 Apr 2024`. [B1SH0PP](https://github.com/B1SH0PP): EfficientAd: Accurate Visual Anomaly Detection at Millisecond-Level Latencies.- `18 Apr 2024`. [lindsayshuo](https://github.com/lindsayshuo): YOLOv8-p2- `12 Mar 2024`. [lindsayshuo](https://github.com/lindsayshuo): YOLOv8-cls- `11 Mar 2024`. [WuxinrongY](https://github.com/WuxinrongY): YOLOv9: Learning What You Want to Learn Using Programmable Gradient Information- `7 Mar 2024`. [AadeIT](https://github.com/AadeIT): CSRNet: Dilated Convolutional Neural Networks for Understanding the Highly Congested Scenes- `17 Oct 2023`. [Rex-LK](https://github.com/Rex-LK): YOLOv8-Seg- `30 Jun 2023`. [xiaocao-tian](https://github.com/xiaocao-tian) and [lindsayshuo](https://github.com/lindsayshuo): YOLOv8- `1 Mar 2023`. [Nengwp](https://github.com/nengwp): [RCNN](./rcnn) and [UNet](./unet) upgrade to support TensorRT 8.- `18 Dec 2022`. [YOLOv5](./yolov5) upgrade to support v7.0, including instance segmentation.- `12 Dec 2022`. [East-Face](https://github.com/East-Face): [UNet](./unet) upgrade to support v3.0 of [Pytorch-UNet](https://github.com/milesial/Pytorch-UNet).- `26 Oct 2022`. [ausk](https://github.com/ausk): YoloP(You Only Look Once for Panopitic Driving Perception).- `19 Sep 2022`. [QIANXUNZDL123](https://github.com/QIANXUNZDL123) and [lindsayshuo](https://github.com/lindsayshuo): YOLOv7.- `7 Sep 2022`. [xiang-wuu](https://github.com/xiang-wuu): YOLOv5 v6.2 classification models.## Tutorials- [How to make contribution](./tutorials/contribution.md)- [Install the dependencies.](./tutorials/install.md)- [A guide for quickly getting started, taking lenet5 as a demo.](./tutorials/getting_started.md)- [The .wts file content format](./tutorials/getting_started.md#the-wts-content-format)- [Frequently Asked Questions (FAQ)](./tutorials/faq.md)- [Migrating from TensorRT 4 to 7](./tutorials/migrating_from_tensorrt_4_to_7.md)- [How to implement multi-GPU processing, taking YOLOv4 as example](./tutorials/multi_GPU_processing.md)- [Check if Your GPU support FP16/INT8](./tutorials/check_fp16_int8_support.md)- [How to Compile and Run on Windows](./tutorials/run_on_windows.md)- [Deploy YOLOv4 with Triton Inference Server](https://github.com/isarsoft/yolov4-triton-tensorrt)- [From pytorch to trt step by step, hrnet as example(Chinese)](./tutorials/from_pytorch_to_trt_stepbystep_hrnet.md)## Test Environment1. TensorRT 7.x2. TensorRT 8.x(Some of the models support 8.x)## How to runEach folder has a readme inside, which explains how to run the models inside.## ModelsFollowing models are implemented.|Name | Description ||-|-||[mlp](./mlp) | the very basic model for starters, properly documented ||[lenet](./lenet) | the simplest, as a "hello world" of this project ||[alexnet](./alexnet)| easy to implement, all layers are supported in tensorrt ||[googlenet](./googlenet)| GoogLeNet (Inception v1) ||[inception](./inception)| Inception v3, v4 ||[mnasnet](./mnasnet)| MNASNet with depth multiplier of 0.5 from the paper ||[mobilenet](./mobilenet)| MobileNet v2, v3-small, v3-large ||[resnet](./resnet)| resnet-18, resnet-50 and resnext50-32x4d are implemented ||[senet](./senet)| se-resnet50 ||[shufflenet](./shufflenetv2)| ShuffleNet v2 with 0.5x output channels ||[squeezenet](./squeezenet)| SqueezeNet 1.1 model ||[vgg](./vgg)| VGG 11-layer model ||[yolov3-tiny](./yolov3-tiny)| weights and pytorch implementation from [ultralytics/yolov3](https://github.com/ultralytics/yolov3) ||[yolov3](./yolov3)| darknet-53, weights and pytorch implementation from [ultralytics/yolov3](https://github.com/ultralytics/yolov3) ||[yolov3-spp](./yolov3-spp)| darknet-53, weights and pytorch implementation from [ultralytics/yolov3](https://github.com/ultralytics/yolov3) ||[yolov4](./yolov4)| CSPDarknet53, weights from [AlexeyAB/darknet](https://github.com/AlexeyAB/darknet#pre-trained-models), pytorch implementation from [ultralytics/yolov3](https://github.com/ultralytics/yolov3) ||[yolov5](./yolov5)| yolov5 v1.0-v7.0 of [ultralytics/yolov5](https://github.com/ultralytics/yolov5), detection, classification and instance segmentation ||[yolov7](./yolov7)| yolov7 v0.1, pytorch implementation from [WongKinYiu/yolov7](https://github.com/WongKinYiu/yolov7) ||[yolov8](./yolov8)| yolov8, pytorch implementation from [ultralytics/ultralytics](https://github.com/ultralytics/ultralytics) ||[yolov9](./yolov9)| The Pytorch implementation is [WongKinYiu/yolov9](https://github.com/WongKinYiu/yolov9). ||[yolop](./yolop)| yolop, pytorch implementation from [hustvl/YOLOP](https://github.com/hustvl/YOLOP) ||[retinaface](./retinaface)| resnet50 and mobilnet0.25, weights from [biubug6/Pytorch_Retinaface](https://github.com/biubug6/Pytorch_Retinaface) ||[arcface](./arcface)| LResNet50E-IR, LResNet100E-IR and MobileFaceNet, weights from [deepinsight/insightface](https://github.com/deepinsight/insightface) ||[retinafaceAntiCov](./retinafaceAntiCov)| mobilenet0.25, weights from [deepinsight/insightface](https://github.com/deepinsight/insightface), retinaface anti-COVID-19, detect face and mask attribute ||[dbnet](./dbnet)| Scene Text Detection, weights from [BaofengZan/DBNet.pytorch](https://github.com/BaofengZan/DBNet.pytorch) ||[crnn](./crnn)| pytorch implementation from [meijieru/crnn.pytorch](https://github.com/meijieru/crnn.pytorch) ||[ufld](./ufld)| pytorch implementation from [Ultra-Fast-Lane-Detection](https://github.com/cfzd/Ultra-Fast-Lane-Detection), ECCV2020 ||[hrnet](./hrnet)| hrnet-image-classification and hrnet-semantic-segmentation, pytorch implementation from [HRNet-Image-Classification](https://github.com/HRNet/HRNet-Image-Classification) and [HRNet-Semantic-Segmentation](https://github.com/HRNet/HRNet-Semantic-Segmentation) ||[psenet](./psenet)| PSENet Text Detection, tensorflow implementation from [liuheng92/tensorflow_PSENet](https://github.com/liuheng92/tensorflow_PSENet) ||[ibnnet](./ibnnet)| IBN-Net, pytorch implementation from [XingangPan/IBN-Net](https://github.com/XingangPan/IBN-Net), ECCV2018 ||[unet](./unet)| U-Net, pytorch implementation from [milesial/Pytorch-UNet](https://github.com/milesial/Pytorch-UNet) ||[repvgg](./repvgg)| RepVGG, pytorch implementation from [DingXiaoH/RepVGG](https://github.com/DingXiaoH/RepVGG) ||[lprnet](./lprnet)| LPRNet, pytorch implementation from [xuexingyu24/License_Plate_Detection_Pytorch](https://github.com/xuexingyu24/License_Plate_Detection_Pytorch) ||[refinedet](./refinedet)| RefineDet, pytorch implementation from [luuuyi/RefineDet.PyTorch](https://github.com/luuuyi/RefineDet.PyTorch) ||[den

用户评论 (0)

发表评论

您的昵称:

评分

评论内容:

验证码:

tensorrt yolo系列转trt模型plugin

资源文件列表(大概)

资源内容介绍

用户评论 (0)

发表评论

相关资源

golang支付合集（包括微信支付，支付宝，后期会加入聚合支付）.zip

代码合集(3).zip

【微信小程序源码期末大作业毕业设计】仿网易蜗牛读书-小程序项目源码.zip

PHP 和 MySQL 构建的高效简洁的内容管理系统.zip

tensorrt yolo系列 转trt模型plugin

资源文件列表(大概)

资源内容介绍

用户评论 (0)

发表评论

相关资源

golang支付合集（包括微信支付，支付宝，后期会加入聚合支付）.zip

代码合集(3).zip

【微信小程序源码期末大作业毕业设计】仿网易蜗牛读书-小程序项目源码.zip

PHP 和 MySQL 构建的高效简洁的内容管理系统.zip

tensorrt yolo系列转trt模型plugin