void ImageProcessor::setParams()

in awstreamer/gst_plugins/cvmlfilter/ImageProcessor.cpp [29:61]


void ImageProcessor::setParams(const std::map<std::string, std::string> &params) {
    // 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"));
    }
}