File tree 2 files changed +12
-2
lines changed
2 files changed +12
-2
lines changed Original file line number Diff line number Diff line change 46
46
./yolox < path/to/your/engine_file> -i < path/to/image>
47
47
```
48
48
49
+ NOTE: for ` trtexec ` users, modify ` INPUT_BLOB_NAME ` and ` OUTPUT_BLOB_NAME ` as the following code.
50
+ ```
51
+ const char* INPUT_BLOB_NAME = "images";
52
+ const char* OUTPUT_BLOB_NAME = "output";
53
+ ```
54
+
55
+ Here is the command to convert the small onnx model to tensorrt engine file:
56
+ ```
57
+ trtexec --onnx=yolox_s.onnx --saveEngine=yolox_s.trt
58
+ ```
Original file line number Diff line number Diff line change @@ -31,8 +31,8 @@ using namespace nvinfer1;
31
31
static const int INPUT_W = 640 ;
32
32
static const int INPUT_H = 640 ;
33
33
static const int NUM_CLASSES = 80 ;
34
- const char * INPUT_BLOB_NAME = " images " ;
35
- const char * OUTPUT_BLOB_NAME = " output " ;
34
+ const char * INPUT_BLOB_NAME = " input_0 " ;
35
+ const char * OUTPUT_BLOB_NAME = " output_0 " ;
36
36
static Logger gLogger ;
37
37
38
38
cv::Mat static_resize (cv::Mat& img) {
You can’t perform that action at this time.
0 commit comments