in awstreamer/gst_plugins/cvmlfilter/ImageProcessor.cpp [29:61]
void ImageProcessor::setParams(const std::map<std::string, std::string> ¶ms) {
// Parse string to useful parameters
if (!params.empty()) {
for (auto it = params.begin(); it != params.end(); ++it) {
GST_INFO("\t%s: %s", it->first.c_str(), it->second.c_str());
}
}
// Load CVML model
if (params.find(CONFIG_CVML_MODEL_NAME) != params.end() &&
params.find(CONFIG_CVML_MODEL_PARAMS) != params.end()) {
GST_INFO("CVML model loaded: %s", params.at(CONFIG_CVML_MODEL_NAME).c_str());
cvNet = cv::dnn::readNetFromTensorflow(
params.at(CONFIG_CVML_MODEL_NAME), params.at(CONFIG_CVML_MODEL_PARAMS));
}
// Face cascade
if (params.find("face_cascade") != params.end()) {
if (!face_cascade.load(params.at("face_cascade"))) {
GST_ERROR("--(!)Error loading face cascade: %s\n", params.at("face_cascade").c_str());
}
}
// Pipeline action
if (params.find("pipeline_action") != params.end()) {
default_action = (PipelineAction)std::stoi(params.at("pipeline_action"));
}
// Threshold
if (params.find("threshold") != params.end()) {
threshold = std::stoi(params.at("threshold"));
}
}