diff --git a/doc/detection.md b/doc/detection.md index 47c85d78e2fbd41ec4a118be15e0d347e6c70fbb..71034876b42efff5053f36e5ea21f6652356063e 100644 --- a/doc/detection.md +++ b/doc/detection.md @@ -79,10 +79,10 @@ python3 tools/eval.py -c configs/det/det_mv3_db.yml -o Global.checkpoints="./ou 测试单张图像的检测效果 ``` -python3 tools/infer_det.py -c config/det/det_mv3_db.yml -o TestReader.single_img_path="./doc/imgs_en/img_10.jpg" +python3 tools/infer_det.py -c config/det/det_mv3_db.yml -o TestReader.single_img_path="./doc/imgs_en/img_10.jpg" Global.checkpoints="./output/det_db/best_accuracy" ``` 测试文件夹下所有图像的检测效果 ``` -python3 tools/infer_det.py -c config/det/det_mv3_db.yml -o TestReader.single_img_path="./doc/imgs_en/" +python3 tools/infer_det.py -c config/det/det_mv3_db.yml -o TestReader.single_img_path="./doc/imgs_en/" Global.checkpoints="./output/det_db/best_accuracy" ``` diff --git a/tools/eval.py b/tools/eval.py index 949f3a343e6f512995363d60c7b6cd3d2e9535f3..304a13401d619fad65984f956357e02581383e18 100755 --- a/tools/eval.py +++ b/tools/eval.py @@ -72,12 +72,12 @@ def main(): init_model(config, eval_program, exe) if alg in ['EAST', 'DB']: - eval_reader = reader_main(config=config, mode="test") + eval_reader = reader_main(config=config, mode="eval") eval_info_dict = {'program':eval_program,\ 'reader':eval_reader,\ 'fetch_name_list':eval_fetch_name_list,\ 'fetch_varname_list':eval_fetch_varname_list} - metrics = eval_det_run(exe, config, eval_info_dict, "test") + metrics = eval_det_run(exe, config, eval_info_dict, "eval") else: reader_type = config['Global']['reader_yml'] if "benchmark" not in reader_type: