-
Notifications
You must be signed in to change notification settings - Fork 314
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
有关前向推断的问题 #46
Comments
1.可以放到最前面 |
@enazoe yolo-tensorrt/modules/yolo.cpp Line 177 in 91455c6
的后面, 即: 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. |
@enazoe |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
作者您好,
1.在创建engine的过程
我看您是把所有层都解析完成之后, 才判断是否有已经产生好的engine. 可不可以把这个判断放在最前面?
yolo-tensorrt/modules/yolo.cpp
Line 475 in 91455c6
2.关于前向推断
我想直接读取保存好的engine 然后进行推断. 也就是直接
而不进行初始化去创建engine
但是这样做是有错的
请问这里我应该如何实现比较好?
感谢!
The text was updated successfully, but these errors were encountered: