Michel Promonet commited on
Commit
a84fcb7
·
1 Parent(s): a13db33

fix(trt): update name to work with trtexec (#1786)

Browse files
Files changed (1) hide show
  1. demo/TensorRT/cpp/yolox.cpp +2 -2
demo/TensorRT/cpp/yolox.cpp CHANGED
@@ -31,8 +31,8 @@ using namespace nvinfer1;
31
  static const int INPUT_W = 640;
32
  static const int INPUT_H = 640;
33
  static const int NUM_CLASSES = 80;
34
- const char* INPUT_BLOB_NAME = "input_0";
35
- const char* OUTPUT_BLOB_NAME = "output_0";
36
  static Logger gLogger;
37
 
38
  cv::Mat static_resize(cv::Mat& img) {
 
31
  static const int INPUT_W = 640;
32
  static const int INPUT_H = 640;
33
  static const int NUM_CLASSES = 80;
34
+ const char* INPUT_BLOB_NAME = "images";
35
+ const char* OUTPUT_BLOB_NAME = "output";
36
  static Logger gLogger;
37
 
38
  cv::Mat static_resize(cv::Mat& img) {