bevdet-tensorrt-cpp icon indicating copy to clipboard operation
bevdet-tensorrt-cpp copied to clipboard

one分支导模型出现问题`BEVDepth4DTRT is not in the models registry`

Open linClubs opened this issue 1 year ago • 2 comments

您好,感谢您的工作。我在使用one分支测试bevdet.yaml,运行时YAML:会报错,我看了代码发现是bevdet.yaml中有些参数没有给出来,我测试bevdet_lt_depth.yaml配置时,使用tools/convert_bevdet_to_TRT.py脚本将pth转engine时,会出现KeyError: 'BEVDepth4DTRT is not in the models registry'错误,转模型使用的分支是BEVDetv2.1。我应该怎么处理才能使bevdet-r50-4dlongterm-depth-cbgs.pth文件转成engine文件?谢谢

linClubs avatar Sep 15 '23 01:09 linClubs