Commit cf42adab authored by zhaoyunfei's avatar zhaoyunfei

再次修改参数

parent 0ef776fc
...@@ -85,12 +85,12 @@ DetectAnomaly::DetectAnomaly() ...@@ -85,12 +85,12 @@ DetectAnomaly::DetectAnomaly()
// // INPUT_RESIZE = 0.25; // // INPUT_RESIZE = 0.25;
cfgAnomaly.input_resize = 0.25; cfgAnomaly.input_resize = 0.25;
cfgAnomaly.owner_search_roi = 10; cfgAnomaly.owner_search_roi = 10;
cfgAnomaly.gmm_learn_frame = 300; cfgAnomaly.gmm_learn_frame = 600;
cfgAnomaly.max_sfg = 15625; cfgAnomaly.max_sfg = 15625;
cfgAnomaly.min_sfg = 15; cfgAnomaly.min_sfg = 15;
cfgAnomaly.max_fg = 75000; cfgAnomaly.max_fg = 75000;
cfgAnomaly.min_fg = 50; cfgAnomaly.min_fg = 50;
cfgAnomaly.buffer_length = 100; cfgAnomaly.buffer_length = 200;
cfgAnomaly.image_height = 770 * cfgAnomaly.input_resize; cfgAnomaly.image_height = 770 * cfgAnomaly.input_resize;
cfgAnomaly.image_width = 2400 * cfgAnomaly.input_resize; cfgAnomaly.image_width = 2400 * cfgAnomaly.input_resize;
...@@ -101,12 +101,12 @@ DetectAnomaly::DetectAnomaly() ...@@ -101,12 +101,12 @@ DetectAnomaly::DetectAnomaly()
// set gloabl var // set gloabl var
INPUT_RESIZE = 1.0; INPUT_RESIZE = 1.0;
OWNER_SEARCH_ROI = 10; OWNER_SEARCH_ROI = 10;
GMM_LEARN_FRAME = 300; //frame number for GMM initialization GMM_LEARN_FRAME = 600; //frame number for GMM initialization
MAX_SFG = 15625; MAX_SFG = 15625;
MIN_SFG = 15; MIN_SFG = 15;
MAX_FG = 75000; MAX_FG = 75000;
MIN_FG = 50; MIN_FG = 50;
BUFFER_LENGTH = 100; BUFFER_LENGTH = 200;
GMM_LONG_LEARN_RATE = 0.0001; GMM_LONG_LEARN_RATE = 0.0001;
GMM_SHORT_LEARN_RATE = 0.002; GMM_SHORT_LEARN_RATE = 0.002;
// INPUT_RESIZE = 0.25; // INPUT_RESIZE = 0.25;
......
...@@ -31,8 +31,8 @@ void InitCnrt(bool useConfig,std::vector<std::pair<std::string,std::string>> mod ...@@ -31,8 +31,8 @@ void InitCnrt(bool useConfig,std::vector<std::pair<std::string,std::string>> mod
// } // }
} }
else{ else{
config_v3.file_model_cfg = "/media/6542bc52-3d7a-4bea-b9b6-f9343ca8560f/Abandoned_detection_20210122/yolov3/configs/yolov3-spp_1batch.cfg"; config_v3.file_model_cfg = "/media/nvidia/Data/whl/yolo-tensorrt/configs/yolov3-spp_1batch.cfg";
config_v3.enginepath = "/media/6542bc52-3d7a-4bea-b9b6-f9343ca8560f/Abandoned_detection_20210122/yolov3/configs/yolov3-spp_final-kINT8-batch1.engine"; config_v3.enginepath = "/media/nvidia/Data/whl/yolo-tensorrt/configs/yolov3-spp_final-kINT8-batch1.engine";
} }
detector->init(config_v3); detector->init(config_v3);
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment