Michel Promonet
commited on
Commit
·
a84fcb7
1
Parent(s):
a13db33
fix(trt): update name to work with trtexec (#1786)
Browse files
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 = "
|
35 |
-
const char* OUTPUT_BLOB_NAME = "
|
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) {
|