bevdet-tensorrt-cpp
bevdet-tensorrt-cpp copied to clipboard
one分支导模型出现问题`BEVDepth4DTRT is not in the models registry`
您好,感谢您的工作。我在使用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文件?谢谢