Skip to content

Commit

Permalink
fix(trt): update name to work with trtexec (#1786)
Browse files Browse the repository at this point in the history 
fix(trt): update name to work with trtexec (#1786)
  • Loading branch information
@michelpromonet
michelpromonet committed Jul 10, 2024
1 parent ac58e0a commit a242a05
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions demo/TensorRT/cpp/yolox.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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) {
Expand Down

1 comment on commit a242a05

@starlipernl
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This change breaks the ability to use torch2trt converted models with this demo script.

Please sign in to comment.