coding

Scheaven

2021-09-08 9b1532d86c2cf48a63017f3460897d8d14b98b60

 src/detecter_tools/detector.cpp

@@ -31,10 +31,10 @@
        vec_ds_images.emplace_back(img, _vec_net_type[_config.net_type], _p_net->getInputH(), _p_net->getInputW());
    }
    cv::Mat trtInput = blobFromDsImages(vec_ds_images, _p_net->getInputH(),_p_net->getInputW());
    timer.out("pre");
    timer.out("eve pre detect ");

    _p_net->doInference(trtInput.data, vec_ds_images.size());
    timer.reset();
    _p_net->doInference(trtInput.data, vec_ds_images.size());
    for (uint32_t i = 0; i < vec_ds_images.size(); ++i)
    {
        auto curImage = vec_ds_images.at(i);
@@ -65,9 +65,8 @@
        }
        vec_batch_result[i] = vec_result;
    }
    timer.out("post");
    timer.out("eve pre detect post");
    DEBUG("--detect over--" );

}

void Detector::set_gpu_id(const int id)
@@ -95,9 +94,9 @@

void Detector::build_net()
{
    if(_config.net_type == SMALL)
        _p_net = std::unique_ptr<Detecter>{new Detecter(_info,_infer_param,2)};
    else{
    if(_config.net_type == COMMON)
        _p_net = std::unique_ptr<Detecter>{new Detecter(_info,_infer_param,1)};
    else{
        _p_net = std::unique_ptr<Detecter>{new Detecter(_info,_infer_param,2)};
    }
}