qq_35914625的博客

Pytorch pb文件转TensorRT 7.0 版本前言环境要求转换步骤githubpb到onnxonnx转trt预测代码 前言 tensorRT的加速效果还是挺明显的。我主要提供两个版本的代码，一个是分类resnet50模型，一个是目标检测yolo的版本。 环境要求 Pytorch==1.2.0 CUDA 10.2 TensorRT 7.0 转换步骤 pb->onnx(caffe)->trt(tensorRT) github https://github.com/yanjingke/ten

总结pytorch语法 1.scatter_(input, dim, index, src) >>> x = torch.rand(2, 5) >>> x 0.4319 0.6500 0.4080 0.8760 0.2355 0.2609 0.4711 0.8486 0.8573 0.1029 [torch.FloatTensor of size 2x5] LongTensor的shape刚好与x的shape对应，也就是LongTensor每个in