mingj2021
mingj2021
firstly ,change opset_version(14,15,16,17), and try again
build with docker please, the repos has been strictly tested in docker.on other dev environment, only simple test.
试试 sam_vit_l ,windows 环境 没进行严格测试,建议docker 环境,本工程提供转化原理及代码,删除 预处理和后处理部分,即可,你可以尝试下,这个网络也不是很复杂的,不在维护了,抱歉了各位!!!
可以看看readme.md 文件,有教程的,删除预处理和后处理代码,关注input_dim 维度问题,网络导出流程都是这种思路,python端学习我的思路,以及c++端,预处理和后处理封装代码,有问题在联系我吧!!!预祝你修炼成功。
orig_im_size 这个参数不需要啊,
这个参数 跟后处理有关,在模型导出前,我把这部分代码在 forward中删除了,你可以试试或者想一想,为什么要删除。这个工程我不在维护了,好久了,没什么人气,本意也是介绍模型转化的常规思路。
orig_im_size ,跟这个变量相关的代码,涉及到动态分配内存问题,而c++中 trt 需要提前为 input output分配内存的(创建上下文context时), 当trt 执行到跟这变量这块相关代码时,就会报错的,所以在forward函数中,删除这段代码,并移植到 后处理部分。
可以自己把这个参数加进去,调试看看,动手实现,预处理 和后处理 这段代码,任何模型都绕不开这块的,建议平时多练练哈!
have you built successfully?