| 101 | } |
| 102 | |
| 103 | void yolo::YOLO::check() |
| 104 | { |
| 105 | int idx; |
| 106 | nvinfer1::Dims dims; |
| 107 | |
| 108 | sample::gLogInfo << "the engine's info:" << std::endl; |
| 109 | for (auto layer_name : m_param.input_output_names) |
| 110 | { |
| 111 | idx = this->m_engine->getBindingIndex(layer_name.c_str()); |
| 112 | dims = this->m_engine->getBindingDimensions(idx); |
| 113 | sample::gLogInfo << "idx = " << idx << ", " << layer_name << ": "; |
| 114 | for (int i = 0; i < dims.nbDims; i++) |
| 115 | { |
| 116 | sample::gLogInfo << dims.d[i] << ", "; |
| 117 | } |
| 118 | sample::gLogInfo << std::endl; |
| 119 | } |
| 120 | sample::gLogInfo << "the context's info:" << std::endl; |
| 121 | for (auto layer_name : m_param.input_output_names) |
| 122 | { |
| 123 | idx = this->m_engine->getBindingIndex(layer_name.c_str()); |
| 124 | dims = this->m_context->getBindingDimensions(idx); |
| 125 | sample::gLogInfo << "idx = " << idx << ", " << layer_name << ": "; |
| 126 | for (int i = 0; i < dims.nbDims; i++) |
| 127 | { |
| 128 | sample::gLogInfo << dims.d[i] << ", "; |
| 129 | } |
| 130 | sample::gLogInfo << std::endl; |
| 131 | } |
| 132 | } |
| 133 | void yolo::YOLO::copy(const std::vector<cv::Mat>& imgsBatch) |
| 134 | { |
| 135 | #if 0 |