thebears/deepstream_yolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update RT-DETR and YOLOX

Browse Source
This commit is contained in:
Marcos Luciano
2024-11-14 23:52:05 -03:00
parent f5c486cb7f
commit 0cc84af9b9
7 changed files with 50 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
nvdsinfer_custom_impl_Yolo/yolo.cpp
View File

@@ -146,9 +146,9 @@ Yolo::createEngine(nvinfer1::IBuilder* builder)
"config_infer file to get better accuracy\n" << std::endl;
}
}
if (m_ClusterMode != 2) {
std::cout << "NOTE: Wrong cluster-mode is set, make sure to set cluster-mode=2 on the config_infer file\n" <<
std::endl;
if (m_ClusterMode != 2 && m_ClusterMode != 4) {
std::cout << "NOTE: Wrong cluster-mode is set, make sure to set cluster-mode=4 (RT-DETR or custom NMS) or " <<
"cluster-mode=2 on the config_infer file\n" << std::endl;
}
if (m_NetworkMode == "FP16") {