优化仓库

孙天宇

2022-07-12 ce9d187fd294cca192a27f52719094e9df7b1b62

 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)