Update yolox.cpp: fix INPUT_BLOB_NAME & OUTPUT_BLOB_NAME

Megvii-BaseDetection · Jun 21, 2024 · 1e11519 · 1e11519
1 parent ac58e0a
commit 1e11519
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/demo/TensorRT/cpp/yolox.cpp b/demo/TensorRT/cpp/yolox.cpp
@@ -31,8 +31,8 @@ using namespace nvinfer1;
 static const int INPUT_W = 640;
 static const int INPUT_H = 640;
 static const int NUM_CLASSES = 80;
-const char* INPUT_BLOB_NAME = "input_0";
-const char* OUTPUT_BLOB_NAME = "output_0";
+const char* INPUT_BLOB_NAME = "images";
+const char* OUTPUT_BLOB_NAME = "output";
 static Logger gLogger;
 
 cv::Mat static_resize(cv::Mat& img) {