有关前向推断的问题 about yolo-tensorrt

[enazoe]

1.可以放到最前面
2.程序会根据你的模型自动去找对应的engine，如果没有才会生成engin，有的话会直接加载。再就是engin文件是和平台绑定的并不是所有平台通用的

[beizhengren]

@enazoe
作者您好, 我放到了

yolo-tensorrt/modules/yolo.cpp

Line 177 in 91455c6

trtWeights.push_back(divWeights);

的后面, 即:

    trtWeights.push_back(divWeights);

    if (fileExists(m_EnginePath))
    {
        std::cout << "Using previously generated plan file located at " << m_EnginePath
                  << std::endl;
        destroyNetworkUtils(trtWeights);
        return;
    }

会报以下错误:

ERROR: yolo-trt: ~/yolo-tensorrt/modules/yolo.cpp:578: 
void Yolo::allocateBuffers(): Assertion `(tensor.bindingIndex != -1) && "Invalid output binding index"' failed.

[beizhengren]

@enazoe
作者您好,我找到原因了,是因为对输出的内存分配和确认问题导致的.
现在解决了,谢谢~