weixin_45415546的博客

升级到Tensort7, ONNX模型创建网络和预测都需要用V2版本。创建模型 const auto explicitBatch = 1U << static_cast<uint32_t>(NetworkDefinitionCreationFlag::kEXPLICIT_BATCH); INetworkDefinition *network = builder->createNetworkV2(explicitBatch); // 创建TensorRT网络

cafffe是没有自适用池化层,在不写插件层的情况下只能转化为avgpooladaptive层的信息只有input_size 和output_size这时候需要倒回去算stride kernel_size正向计算output_size = ceil((input_size+2∗padding−kernel_size)/stride)+1反向计算padding设置为0stride ...