漂亮汽车

文章
5
资源
0
加入时间
3年0月9天

FCN8s 转 tensorrt(通过wts文件)

先用pytorch训练FCN,然后把权重的pth文件,先转为wts文件,然后在NX板上转为tensorrt的engine文件只写具体思路和遇到的问题,完整版参考github这里写的是paper里的由vgg16而来的FCN,首先实现vgg16的tensorrt先来看下vgg16的结构'vgg16': [64, 64, 'M', 128, 128, 'M', 256, 256, 256, 'M', 512, 512, 512, 'M', 512, 512, 512, 'M']其中每一层的数字表示输出