-
Notifications
You must be signed in to change notification settings - Fork 315
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
trt8 maxpool的問題 #188
Comments
抱歉 我覺得問題也不在這 我再查查看 不過真是沒道理 trt7 ok trt8則失敗 |
sorry, 我應該確定是maxpool的問題,因在建enging時(以下我只列部分) WARNING: Weights [name=conv_11 + batch_norm_11.weight] had the following issues when converted to FP16: 列config時沒錯 建enging時沒有13(maxpool) |
这个升级到tensorrt8后NvInfer.h中的这个class TRT_DEPRECATED IOutputDimensionsFormula被遗弃了,所以m_TinyMaxpoolPaddingFormula也被注释掉了,我看这个代码的readme里也把对yolov tiny的支持去掉了,不知道后续用什么替代方案来支持呢? |
我前面的問題我發現是在trt8 的yolo.cpp-line 410 被comment掉, trt7則沒有comment
if (m_configBlocks.at(i).at("size") == "2" && m_configBlocks.at(i).at("stride") == "1")
{
// m_TinyMaxpoolPaddingFormula->addSamePaddingLayer("maxpool_" + std::to_string(i));
}
而我用了
[maxpool]
size=2
stride=2
在trt7成功偵測
在trt8則失敗 ( trt8 remove "setPoolingOutputDimensionsFormula")
請問可以解決這個問題嗎?
謝謝你花寶貴的時間看我的問題.
The text was updated successfully, but these errors were encountered: