PaddleOCR2Pytorch icon indicating copy to clipboard operation
PaddleOCR2Pytorch copied to clipboard

Published 20 hours ago • verified publisher icon frotms

pytorchocr_utility.py里缺少PSE检测模型的network_config

Open CallMeFrozenBanana opened this issue 2 years ago • 1 comments

但外面det_algorithm又是有PSE这一项的

CallMeFrozenBanana avatar Nov 16 '22 07:11 CallMeFrozenBanana

原来是需要特别指定yml，具体如下：

python ./tools/infer/predict_system.py --image_dir ./doc/imgs_bd --det_model_path ./pretrained/det_mv3_pse_v2.0_infer.pth --rec_model_path ./pretrained/ch_ptocr_v3_rec_infer.pth --use_angle_cls 1 --cls_model_path ./pretrained/ch_ptocr_mobile_v2.0_cls_infer.pth --det_algorithm PSE --det_yaml_path ./configs/det/det_mv3_pse.yml

python ./tools/infer/predict_det.py --det_model_path ./pretrained/det_mv3_pse_v2.0_infer.pth --image_dir ./doc/imgs_bd --det_algorithm PSE --det_yaml_path ./configs/det/det_mv3_pse.yml

CallMeFrozenBanana avatar Nov 16 '22 07:11 CallMeFrozenBanana