PaddleOCR sast算法检测模型pretrained模型与inference模型效果不一致

x4shl7ld 于 2022-10-22 发布在其他

关注(0)|答案(8)|浏览(358)

操作步骤如下：
1，sast pretrained 模型使用命令预测：
python tools/infer_det.py -c configs/det/det_r50_vd_sast_icdar15.yml -o Global.infer_img="" Global.pretrained_model="output/sast_r50_vd_ic15_train/best_accuracy"

2，得到结果：

3，使用命令转换inference 模型：
python tools/export_model.py -c configs/det/det_r50_vd_sast_icdar15_train.yml -o Global.pretrained_model=output/sast_r50_vd_ic15_train/best_accuracy Global.save_inference_dir=inference/sast_r50_vd_ic15_train

4，使用命令推理：
python tools/infer/predict_det.py --det_algorithm="SAST" --image_dir="" --det_model_dir="inference/sast_r50_vd_ic15_train"

5，得到结果：

这个问题真的没有办法解决吗？

ps:附上det_r50_vd_sast_icdar15.yml配置文件关键部分:

paddleocr

来源：https://github.com/PaddlePaddle/PaddleOCR/issues/3225

8条答案

按热度按时间

68bkxrlz1#

对齐一下预测的后处理参数试试：
PaddleOCR/tools/infer/utility.py

Lines 57 to 60 in 39161f5

| | # SAST parmas |
| | parser.add_argument("--det_sast_score_thresh", type=float, default=0.5) |
| | parser.add_argument("--det_sast_nms_thresh", type=float, default=0.2) |
| | parser.add_argument("--det_sast_polygon", type=bool, default=False) |

赞(0）回复(0）举报 2022-10-22

1l5u6lss2#

对齐一下预测的后处理参数试试：
PaddleOCR/tools/infer/utility.py

Lines 57 to 60 in 39161f5

| | # SAST parmas |
| | parser.add_argument("--det_sast_score_thresh", type=float, default=0.5) |
| | parser.add_argument("--det_sast_nms_thresh", type=float, default=0.2) |
| | parser.add_argument("--det_sast_polygon", type=bool, default=False) |

本来就是这样的