tensorrtx
tensorrtx copied to clipboard
对于修改过结构的模型要如何进行转换?
如题,我对yolov5的backbone进行了更换,如果向生成基于这个模型的engine文件应该对tensorrtx的哪部分进行调整?
yolov5.cpp build_engine()
如题,我对yolov5的backbone进行了更换,如果向生成基于这个模型的engine文件应该对tensorrtx的哪部分进行调整?
你好,请问你有没有成功转换修改后的模型啊?可以交流一下吗?