\brief CarsCounting::InitDetector \param frame
| 136 | /// \param frame |
| 137 | /// |
| 138 | bool CarsCounting::InitDetector(cv::UMat frame) |
| 139 | { |
| 140 | if (!m_trackerSettingsLoaded) |
| 141 | return false; |
| 142 | |
| 143 | config_t config; |
| 144 | |
| 145 | config.emplace("modelConfiguration", m_trackerSettings.m_nnConfig); |
| 146 | config.emplace("modelBinary", m_trackerSettings.m_nnWeights); |
| 147 | config.emplace("confidenceThreshold", std::to_string(m_trackerSettings.m_confidenceThreshold)); |
| 148 | config.emplace("classNames", m_trackerSettings.m_classNames); |
| 149 | config.emplace("maxCropRatio", std::to_string(m_trackerSettings.m_maxCropRatio)); |
| 150 | config.emplace("maxBatch", std::to_string(m_trackerSettings.m_maxBatch)); |
| 151 | config.emplace("gpuId", std::to_string(m_trackerSettings.m_gpuId)); |
| 152 | config.emplace("net_type", m_trackerSettings.m_netType); |
| 153 | config.emplace("inference_precision", m_trackerSettings.m_inferencePrecision); |
| 154 | config.emplace("video_memory", std::to_string(m_trackerSettings.m_maxVideoMemory)); |
| 155 | config.emplace("dnnTarget", m_trackerSettings.m_dnnTarget); |
| 156 | config.emplace("dnnBackend", m_trackerSettings.m_dnnBackend); |
| 157 | config.emplace("inWidth", std::to_string(m_trackerSettings.m_inputSize.width)); |
| 158 | config.emplace("inHeight", std::to_string(m_trackerSettings.m_inputSize.height)); |
| 159 | |
| 160 | for (auto wname : m_trackerSettings.m_whiteList) |
| 161 | { |
| 162 | config.emplace("white_list", wname); |
| 163 | } |
| 164 | |
| 165 | m_detector = BaseDetector::CreateDetector((tracking::Detectors)m_trackerSettings.m_detectorBackend, config, frame); |
| 166 | |
| 167 | return m_detector.operator bool(); |
| 168 | } |
| 169 | |
| 170 | /// |
| 171 | /// \brief CarsCounting::InitTracker |