深度学习
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 · 9316 阅读 · 4 评论 -
卷积神经网络计算输出尺寸
1.Convolution out = floor((in+2padding-dilation(kernel_size-1)-1)/stride+1) dilation默认情况为1 out=floor((in+2padding-kernel_size)/stride+1) 2.Pooling out = floor((in+2padding-dilation(kernel_size-1)-1)/s...原创 2019-11-25 09:54:42 · 302 阅读 · 0 评论 -
深度学习caffe之Slice层
Concat是将多个bottom合并成一个top slice层把一个bottom分解成多个top slice层三个参数: axis:默认为1,切分channel维度 slice_dim:过时参数同axis,不支持负数 slice_point:切分点,没有设置的时候则对axis进行均匀分解 message SliceParameter { // The axis along which to s...原创 2019-11-05 15:09:46 · 850 阅读 · 0 评论