自定义博客皮肤VIP专享

*博客头图：

点击选择上传的图片

格式为PNG、JPG，宽度*高度大于1920*100像素，不超过2MB，主视觉建议放在右侧，请参照线上博客头图

请上传大于1920*100像素的图片！

博客底图：

点击选择上传的图片

图片格式为PNG、JPG，不超过1MB，可上下左右平铺至整个背景

栏目图：

点击选择上传的图片

图片格式为PNG、JPG，图片宽度*高度为300*38像素，不超过0.5MB

主标题颜色：

RGB颜色，例如：#AFAFAF

Hover：

RGB颜色，例如：#AFAFAF

副标题颜色：

RGB颜色，例如：#AFAFAF

预览取消提交

自定义博客皮肤

上一步保存

qq_21612131的博客

lyh_robert

博客等级

码龄11年

1
原创

6
点赞

78
收藏

10
粉丝

关注

私信

分类专栏

模型性能优化 1篇
论文解读 1篇

最新评论

【论文解读】用于卷积神经网络的注意力机制(Attention)----CBAM: Convolutional Block Attention Module
m0_46686292: 请问一下我想把vgg加入attention里你这个代码可以使用吗
使用TensorRT加速GPU上的TensorFlow推理（翻译）
qq_43456025: 用tensorrt加速tensorflow时出现问题，显示Failed to create Inputlayer。。我的tensorrt是4.0版本的,本人学生党，菜鸡一枚#(泪) 。在TX2上部署的tensorflow-GPU1.9 用的tensorflow-GPU1.11.0训练的，用1.9时可以正常地把tensorflow1.11.0的ckpt生成pb。到是用tensorrt加速时出现问题。我没有把pb转换为uff。而是直接用的trt.create_inference_graph来推导的。
【论文解读】用于卷积神经网络的注意力机制(Attention)----CBAM: Convolutional Block Attention Module
weixin_39157106: scSENet是并联，这个网络是串联；这个网络的channel pooling令人眼前一亮；
【论文解读】用于卷积神经网络的注意力机制(Attention)----CBAM: Convolutional Block Attention Module
weixin_39157106 回复徐梓航: 这个池化和我们平时看到的池化不一样，比如channel Attention中的池化，其实处理使用tf.nn.avg_pool()外，还可以使用tf.reduce_mean(input_tensor, axis=[1, 2], keepdims=True)来代替，这是对spatial的池化；由此推到，在channel中的池化怎么处理呢？通过tf.nn.avg_pool()不能处理，那就通过tf.reduce_mean(input_tensor, axis=3, keepdims=True)来处理，变达到了池化的效果；
【论文解读】用于卷积神经网络的注意力机制(Attention)----CBAM: Convolutional Block Attention Module
weixin_39157106 回复 Aser_Chen: 用tf.reduce_mean(input_tensor, axis=[1, 2], keepdims=True) 来代替Global avg pooling，同理可以用tf.reduce_max来代替Global max pooling；在使用tf.nn.avg_pool()时，注意ksize要指定明确，不能出现多个None；

模型性能优化

关注

文章平均质量分 83

关注数：文章数：1 文章阅读量：2926 文章收藏量：6

作者: lyh_robert

这个作者很懒，什么都没留下…

展开

使用TensorRT加速GPU上的TensorFlow推理（翻译）

本文翻译于博客Speed up TensorFlow Inference on GPUs with TensorRT，这篇博客介绍了如何使用TensorRT加速TensorFlow模型的推理速度，作者为：Siddharth Sharma — Technical Product Marketing Manager, NVidiaSami Kama — Deep Learning Develop...

翻译 2018-10-19 22:44:09 · 2927 阅读 · 1 评论

模型性能优化

作者: lyh_robert

使用TensorRT加速GPU上的TensorFlow推理（翻译）