### INT8量化和精度测试数据集
+ training RGB images: <https://pan.baidu.com/s/1k7aSSL5MMipWYszlFaBLkA> code: `mst1`
+ validation RGB images: <https://pan.baidu.com/s/1YakbXgBgnhNmYoxySmZaGw>
+ testing RGB images: <https://pan.baidu.com/s/1RXHK64mUfK_GeeoLzqAmeQ>
以上数据来源于:NTIRE 2022 Spectral Reconstruction Challenge的主页:<https://codalab.lisn.upsaclay.fr/competitions/721#participate-get_data>
没有合适的资源?快使用搜索试试~ 我知道了~
人工智能-Transformer-图像建模-首个基于Transformer的图像重建模型MST++的TensorRT模型推断优化
共83个文件
png:38个
py:20个
pyc:11个
1.该资源内容由用户上传,如若侵权请联系客服进行举报
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
版权申诉
5星 · 超过95%的资源 3 下载量 47 浏览量
2021-06-11
15:37:06
上传
评论
收藏 5.38MB ZIP 举报
温馨提示
我们主要采用2条主线优化该网络,TensorRT ONNXParser和TensorRT API两种方式。基于对ONNXParser用Nsight进行Profiling,并最终确定了有针对性的优化方案,包括: 网络中存在大量的Transpose+Reshape操作较为耗时 卷积,分组卷积和转置卷积 Self-Attention 冗余结构耗时 针对于上述问题我们简化网络结构通过TensorRT API重新搭建了网络结构,实现了L2Norm,Self-Attention,LayerNorm,卷积,分组卷积,反卷积和MST++的S_MSA核心结构. L2Norm结构的耗时 LayerNorm结构耗时 Gelu结构耗时
资源推荐
资源详情
资源评论
收起资源包目录
MST-plus-plus-TensorRT-main.zip (83个子文件)
MST-plus-plus-TensorRT-main
performance_int8.py 10KB
data
README.md 429B
NormalizePlugin
NormalizePlugin.cu 21KB
Makefile 1KB
NormalizePlugin.h 6KB
LayerNormPlugin
Makefile.inc 671B
Makefile 446B
testLayerNormPlugin.py 6KB
LayerNormPlugin.cu 3KB
layer_norm.cuh 66KB
LayerNormPlugin.h 7KB
mst_trt_api.py 34KB
architecture
__init__.py 1KB
HDNet.py 13KB
Restormer.py 13KB
MST_Plus_Plus.py 10KB
MIRNet.py 15KB
HSCNN_Plus.py 3KB
edsr.py 2KB
hinet.py 8KB
MPRNet.py 15KB
hrnet.py 24KB
__pycache__
MPRNet.cpython-38.pyc 12KB
HDNet.cpython-38.pyc 12KB
edsr.cpython-38.pyc 3KB
hrnet.cpython-38.pyc 16KB
MST_Plus_Plus.cpython-38.pyc 9KB
HSCNN_Plus.cpython-38.pyc 3KB
Restormer.cpython-38.pyc 11KB
MIRNet.cpython-38.pyc 14KB
hinet.cpython-38.pyc 7KB
MST.cpython-38.pyc 9KB
__init__.cpython-38.pyc 2KB
README.md 103B
MST.py 10KB
performance_latency.py 14KB
get_weights.py 2KB
performance_accuracy.py 17KB
docs
trtapi_layernorm.png 69KB
compare_fig.png 206KB
onnxparser_self_attention.png 65KB
int8_layer.png 36KB
image-20210715140905802.png 13KB
times.png 43KB
image-20210917164119121.png 98KB
image-20210917164140784.png 98KB
image-20220526093800077.png 264KB
selfattention.png 25KB
logo.png 1.67MB
gelu.png 14KB
transpose_reshape.png 40KB
int8_layer3.png 56KB
tr.png 55KB
int8_layer2.png 54KB
abs_error.png 285KB
int8_error.png 109KB
image-20210917164100738.png 98KB
image-20220526093617393.png 16KB
s_msa.png 176KB
int8_acc.png 149KB
Layernorm.png 24KB
image-20220526094439497.png 182KB
image-20220526095217415.png 281KB
latency_vs_throughput.png 99KB
int8_times.png 10KB
image-20220526100055372.png 186KB
trtapi_self_attention.png 77KB
trtapi_l2norm.png 70KB
l2norm.png 14KB
trtapi_gelu.png 62KB
onnxparser_layernorm.png 65KB
onnxparser_gelu.png 58KB
onnxparser_l2norm.png 60KB
rel_error.png 296KB
MST++ TensorRT 2022 美迪康AI Lab.png 174KB
MST++.png 234KB
mst_onnxparser.py 6KB
torch2onnx.py 4KB
model
README.md 330B
requirements.txt 193B
build.sh 6KB
calibrator.py 5KB
mst_config.ini 18KB
共 83 条
- 1
资源评论
- 橘clr2021-06-20用户下载后在一定时间内未进行评价,系统默认好评。
- jlk0152021-06-20用户下载后在一定时间内未进行评价,系统默认好评。
博士僧小星
- 粉丝: 1686
- 资源: 5875
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功