m

suntianyu

2021-08-12 e47cdf6abec38e5249f5258bc7b98824b67eb79f

 src/detecter_tools/detector.cpp

@@ -95,8 +95,8 @@

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