From 4eb29b8ec8777f68c26dfe2a4397cce92110fcfd Mon Sep 17 00:00:00 2001 From: chenshijun <chenshijun@aiotlink.com> Date: 星期三, 08 九月 2021 17:58:20 +0800 Subject: [PATCH] ok --- src/detecter_tools/detector.cpp | 4 ++-- 1 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/detecter_tools/detector.cpp b/src/detecter_tools/detector.cpp index cf3e313..78e136f 100644 --- a/src/detecter_tools/detector.cpp +++ b/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)}; } -- Gitblit v1.8.0