├── LICENSE ├── LayerNormPlugin ├── LayerNormPlugin.cu ├── LayerNormPlugin.h ├── Makefile ├── Makefile.inc ├── layer_norm.cuh └── testLayerNormPlugin.py ├── NormalizePlugin ├── Makefile ├── NormalizePlugin.cu └── NormalizePlugin.h ├── README.md ├── architecture ├── HDNet.py ├── HSCNN_Plus.py ├── MIRNet.py ├── MPRNet.py ├── MST.py ├── MST_Plus_Plus.py ├── README.md ├── Restormer.py ├── __init__.py ├── __pycache__ │ ├── HDNet.cpython-38.pyc │ ├── HSCNN_Plus.cpython-38.pyc │ ├── MIRNet.cpython-38.pyc │ ├── MPRNet.cpython-38.pyc │ ├── MST.cpython-38.pyc │ ├── MST_Plus_Plus.cpython-38.pyc │ ├── Restormer.cpython-38.pyc │ ├── __init__.cpython-38.pyc │ ├── edsr.cpython-38.pyc │ ├── hinet.cpython-38.pyc │ └── hrnet.cpython-38.pyc ├── edsr.py ├── hinet.py └── hrnet.py ├── calibrator.py ├── data └── README.md ├── docs ├── Layernorm.png ├── MST++ TensorRT 2022 美迪康AI Lab.png ├── MST++.png ├── abs_error.png ├── compare_fig.png ├── gelu.png ├── image-20210715140905802.png ├── image-20210917164100738.png ├── image-20210917164119121.png ├── image-20210917164140784.png ├── image-20220526093617393.png ├── image-20220526093800077.png ├── image-20220526094439497.png ├── image-20220526095217415.png ├── image-20220526100055372.png ├── int8_acc.png ├── int8_error.png ├── int8_layer.png ├── int8_layer2.png ├── int8_layer3.png ├── int8_times.png ├── l2norm.png ├── latency_vs_throughput.png ├── logo.png ├── onnxparser_gelu.png ├── onnxparser_l2norm.png ├── onnxparser_layernorm.png ├── onnxparser_self_attention.png ├── rel_error.png ├── s_msa.png ├── selfattention.png ├── times.png ├── tr.png ├── transpose_reshape.png ├── trtapi_gelu.png ├── trtapi_l2norm.png ├── trtapi_layernorm.png └── trtapi_self_attention.png ├── get_weights.py ├── model └── README.md ├── mst_config.ini ├── mst_onnxparser.py ├── mst_trt_api.py ├── performance_accuracy.py ├── performance_int8.py ├── performance_latency.py ├── requirements.txt └── torch2onnx.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/LICENSE -------------------------------------------------------------------------------- /LayerNormPlugin/LayerNormPlugin.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/LayerNormPlugin/LayerNormPlugin.cu -------------------------------------------------------------------------------- /LayerNormPlugin/LayerNormPlugin.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/LayerNormPlugin/LayerNormPlugin.h -------------------------------------------------------------------------------- /LayerNormPlugin/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/LayerNormPlugin/Makefile -------------------------------------------------------------------------------- /LayerNormPlugin/Makefile.inc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/LayerNormPlugin/Makefile.inc -------------------------------------------------------------------------------- /LayerNormPlugin/layer_norm.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/LayerNormPlugin/layer_norm.cuh -------------------------------------------------------------------------------- /LayerNormPlugin/testLayerNormPlugin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/LayerNormPlugin/testLayerNormPlugin.py -------------------------------------------------------------------------------- /NormalizePlugin/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/NormalizePlugin/Makefile -------------------------------------------------------------------------------- /NormalizePlugin/NormalizePlugin.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/NormalizePlugin/NormalizePlugin.cu -------------------------------------------------------------------------------- /NormalizePlugin/NormalizePlugin.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/NormalizePlugin/NormalizePlugin.h -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/README.md -------------------------------------------------------------------------------- /architecture/HDNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/architecture/HDNet.py -------------------------------------------------------------------------------- /architecture/HSCNN_Plus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/architecture/HSCNN_Plus.py -------------------------------------------------------------------------------- /architecture/MIRNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/architecture/MIRNet.py -------------------------------------------------------------------------------- /architecture/MPRNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/architecture/MPRNet.py -------------------------------------------------------------------------------- /architecture/MST.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/architecture/MST.py -------------------------------------------------------------------------------- /architecture/MST_Plus_Plus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/architecture/MST_Plus_Plus.py -------------------------------------------------------------------------------- /architecture/README.md: -------------------------------------------------------------------------------- 1 | ## MST++ 模型 2 | 3 | 该文件夹下的代码来源于： -------------------------------------------------------------------------------- /architecture/Restormer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/architecture/Restormer.py -------------------------------------------------------------------------------- /architecture/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/architecture/__init__.py -------------------------------------------------------------------------------- /architecture/__pycache__/HDNet.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/architecture/__pycache__/HDNet.cpython-38.pyc -------------------------------------------------------------------------------- /architecture/__pycache__/HSCNN_Plus.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/architecture/__pycache__/HSCNN_Plus.cpython-38.pyc -------------------------------------------------------------------------------- /architecture/__pycache__/MIRNet.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/architecture/__pycache__/MIRNet.cpython-38.pyc -------------------------------------------------------------------------------- /architecture/__pycache__/MPRNet.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/architecture/__pycache__/MPRNet.cpython-38.pyc -------------------------------------------------------------------------------- /architecture/__pycache__/MST.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/architecture/__pycache__/MST.cpython-38.pyc -------------------------------------------------------------------------------- /architecture/__pycache__/MST_Plus_Plus.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/architecture/__pycache__/MST_Plus_Plus.cpython-38.pyc -------------------------------------------------------------------------------- /architecture/__pycache__/Restormer.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/architecture/__pycache__/Restormer.cpython-38.pyc -------------------------------------------------------------------------------- /architecture/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/architecture/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /architecture/__pycache__/edsr.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/architecture/__pycache__/edsr.cpython-38.pyc -------------------------------------------------------------------------------- /architecture/__pycache__/hinet.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/architecture/__pycache__/hinet.cpython-38.pyc -------------------------------------------------------------------------------- /architecture/__pycache__/hrnet.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/architecture/__pycache__/hrnet.cpython-38.pyc -------------------------------------------------------------------------------- /architecture/edsr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/architecture/edsr.py -------------------------------------------------------------------------------- /architecture/hinet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/architecture/hinet.py -------------------------------------------------------------------------------- /architecture/hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/architecture/hrnet.py -------------------------------------------------------------------------------- /calibrator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/calibrator.py -------------------------------------------------------------------------------- /data/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/data/README.md -------------------------------------------------------------------------------- /docs/Layernorm.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/Layernorm.png -------------------------------------------------------------------------------- /docs/MST++ TensorRT 2022 美迪康AI Lab.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/MST++ TensorRT 2022 美迪康AI Lab.png -------------------------------------------------------------------------------- /docs/MST++.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/MST++.png -------------------------------------------------------------------------------- /docs/abs_error.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/abs_error.png -------------------------------------------------------------------------------- /docs/compare_fig.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/compare_fig.png -------------------------------------------------------------------------------- /docs/gelu.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/gelu.png -------------------------------------------------------------------------------- /docs/image-20210715140905802.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/image-20210715140905802.png -------------------------------------------------------------------------------- /docs/image-20210917164100738.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/image-20210917164100738.png -------------------------------------------------------------------------------- /docs/image-20210917164119121.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/image-20210917164119121.png -------------------------------------------------------------------------------- /docs/image-20210917164140784.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/image-20210917164140784.png -------------------------------------------------------------------------------- /docs/image-20220526093617393.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/image-20220526093617393.png -------------------------------------------------------------------------------- /docs/image-20220526093800077.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/image-20220526093800077.png -------------------------------------------------------------------------------- /docs/image-20220526094439497.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/image-20220526094439497.png -------------------------------------------------------------------------------- /docs/image-20220526095217415.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/image-20220526095217415.png -------------------------------------------------------------------------------- /docs/image-20220526100055372.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/image-20220526100055372.png -------------------------------------------------------------------------------- /docs/int8_acc.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/int8_acc.png -------------------------------------------------------------------------------- /docs/int8_error.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/int8_error.png -------------------------------------------------------------------------------- /docs/int8_layer.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/int8_layer.png -------------------------------------------------------------------------------- /docs/int8_layer2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/int8_layer2.png -------------------------------------------------------------------------------- /docs/int8_layer3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/int8_layer3.png -------------------------------------------------------------------------------- /docs/int8_times.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/int8_times.png -------------------------------------------------------------------------------- /docs/l2norm.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/l2norm.png -------------------------------------------------------------------------------- /docs/latency_vs_throughput.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/latency_vs_throughput.png -------------------------------------------------------------------------------- /docs/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/logo.png -------------------------------------------------------------------------------- /docs/onnxparser_gelu.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/onnxparser_gelu.png -------------------------------------------------------------------------------- /docs/onnxparser_l2norm.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/onnxparser_l2norm.png -------------------------------------------------------------------------------- /docs/onnxparser_layernorm.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/onnxparser_layernorm.png -------------------------------------------------------------------------------- /docs/onnxparser_self_attention.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/onnxparser_self_attention.png -------------------------------------------------------------------------------- /docs/rel_error.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/rel_error.png -------------------------------------------------------------------------------- /docs/s_msa.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/s_msa.png -------------------------------------------------------------------------------- /docs/selfattention.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/selfattention.png -------------------------------------------------------------------------------- /docs/times.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/times.png -------------------------------------------------------------------------------- /docs/tr.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/tr.png -------------------------------------------------------------------------------- /docs/transpose_reshape.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/transpose_reshape.png -------------------------------------------------------------------------------- /docs/trtapi_gelu.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/trtapi_gelu.png -------------------------------------------------------------------------------- /docs/trtapi_l2norm.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/trtapi_l2norm.png -------------------------------------------------------------------------------- /docs/trtapi_layernorm.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/trtapi_layernorm.png -------------------------------------------------------------------------------- /docs/trtapi_self_attention.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/docs/trtapi_self_attention.png -------------------------------------------------------------------------------- /get_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/get_weights.py -------------------------------------------------------------------------------- /model/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/model/README.md -------------------------------------------------------------------------------- /mst_config.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/mst_config.ini -------------------------------------------------------------------------------- /mst_onnxparser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/mst_onnxparser.py -------------------------------------------------------------------------------- /mst_trt_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/mst_trt_api.py -------------------------------------------------------------------------------- /performance_accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/performance_accuracy.py -------------------------------------------------------------------------------- /performance_int8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/performance_int8.py -------------------------------------------------------------------------------- /performance_latency.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/performance_latency.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/requirements.txt -------------------------------------------------------------------------------- /torch2onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRT2022/MST-plus-plus-TensorRT/HEAD/torch2onnx.py --------------------------------------------------------------------------------