diff --git a/demo/TensorRT/cpp/yolox.cpp b/demo/TensorRT/cpp/yolox.cpp
index ed423380e..e5d1ce73c 100644
--- 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) {