![](https://img-blog.csdnimg.cn/20201014180756918.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
weixin_45415546
这个作者很懒,什么都没留下…
展开
-
TensorRT7 Onnx模型动态多batch问题解决
升级到Tensort7, ONNX模型创建网络和预测都需要用V2版本。创建模型 const auto explicitBatch = 1U << static_cast<uint32_t>(NetworkDefinitionCreationFlag::kEXPLICIT_BATCH); INetworkDefinition *network = builder->createNetworkV2(explicitBatch); // 创建TensorRT网络原创 2020-08-31 10:39:33 · 9244 阅读 · 4 评论 -
卷积神经网络计算输出尺寸
1.Convolutionout = floor((in+2padding-dilation(kernel_size-1)-1)/stride+1)dilation默认情况为1out=floor((in+2padding-kernel_size)/stride+1)2.Poolingout = floor((in+2padding-dilation(kernel_size-1)-1)/s...原创 2019-11-25 09:54:42 · 291 阅读 · 0 评论 -
深度学习caffe之Slice层
Concat是将多个bottom合并成一个topslice层把一个bottom分解成多个topslice层三个参数:axis:默认为1,切分channel维度slice_dim:过时参数同axis,不支持负数slice_point:切分点,没有设置的时候则对axis进行均匀分解message SliceParameter { // The axis along which to s...原创 2019-11-05 15:09:46 · 836 阅读 · 0 评论