m

Scheaven

2021-08-12 d91bfe2d12d0fb5daf2be0a7587045d85a50c4a1

 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)};
    }