|
@@ -291,7 +291,7 @@ class InferImpl : public Infer {
|
|
string engine_file_;
|
|
string engine_file_;
|
|
vector<shared_ptr<trt::Memory<unsigned char>>> preprocess_buffers_;
|
|
vector<shared_ptr<trt::Memory<unsigned char>>> preprocess_buffers_;
|
|
trt::Memory<float> input_buffer_, output_array_;
|
|
trt::Memory<float> input_buffer_, output_array_;
|
|
- trt::Memory<float> classes_indexes_;
|
|
|
|
|
|
+ trt::Memory<float> classes_indices_;
|
|
int network_input_width_, network_input_height_;
|
|
int network_input_width_, network_input_height_;
|
|
Norm normalize_;
|
|
Norm normalize_;
|
|
int num_classes_ = 0;
|
|
int num_classes_ = 0;
|