Commit cf42adab authored by zhaoyunfei's avatar zhaoyunfei

再次修改参数

parent 0ef776fc
......@@ -85,12 +85,12 @@ DetectAnomaly::DetectAnomaly()
// // INPUT_RESIZE = 0.25;
cfgAnomaly.input_resize = 0.25;
cfgAnomaly.owner_search_roi = 10;
cfgAnomaly.gmm_learn_frame = 300;
cfgAnomaly.gmm_learn_frame = 600;
cfgAnomaly.max_sfg = 15625;
cfgAnomaly.min_sfg = 15;
cfgAnomaly.max_fg = 75000;
cfgAnomaly.min_fg = 50;
cfgAnomaly.buffer_length = 100;
cfgAnomaly.buffer_length = 200;
cfgAnomaly.image_height = 770 * cfgAnomaly.input_resize;
cfgAnomaly.image_width = 2400 * cfgAnomaly.input_resize;
......@@ -101,12 +101,12 @@ DetectAnomaly::DetectAnomaly()
// set gloabl var
INPUT_RESIZE = 1.0;
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;
MIN_SFG = 15;
MAX_FG = 75000;
MIN_FG = 50;
BUFFER_LENGTH = 100;
BUFFER_LENGTH = 200;
GMM_LONG_LEARN_RATE = 0.0001;
GMM_SHORT_LEARN_RATE = 0.002;
// INPUT_RESIZE = 0.25;
......
......@@ -31,8 +31,8 @@ void InitCnrt(bool useConfig,std::vector<std::pair<std::string,std::string>> mod
// }
}
else{
config_v3.file_model_cfg = "/media/6542bc52-3d7a-4bea-b9b6-f9343ca8560f/Abandoned_detection_20210122/yolov3/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.file_model_cfg = "/media/nvidia/Data/whl/yolo-tensorrt/configs/yolov3-spp_1batch.cfg";
config_v3.enginepath = "/media/nvidia/Data/whl/yolo-tensorrt/configs/yolov3-spp_final-kINT8-batch1.engine";
}
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