frotms / PaddleOCR2Pytorch

PaddleOCR inference in PyTorch. Converted from [PaddleOCR](https://github.com/PaddlePaddle/PaddleOCR)
Apache License 2.0
809 stars 159 forks source link

cls方向分类模型推理有问题,好像是预处理缺失ClsResizeImg #61

Open williamlzw opened 1 year ago

williamlzw commented 1 year ago

cls方向分类模型推理有问题,好像是预处理缺失,ClsResizeImg

williamlzw commented 1 year ago

ch_ptocr_mobile_v2.0_cls_infer.pth python ./tools/infer/predict_cls.py --cls_model_path D:/paddleocr/ch_ptocr_mobile_v2.0_cls_infer.pth --image_dir ./doc/imgs_words/ch/word_1.jpg --use_angle_cls 1

frotms commented 1 year ago

@williamlzw 请问报什么错误?

williamlzw commented 1 year ago

不同方向图片传进去推理结果只有一个0方向

frotms commented 1 year ago

对比下paddleOCR是否也是一样的吗

williamlzw commented 1 year ago

paddleocr结果也都是0,难道是paddleocr出错了?

frotms commented 1 year ago

可以贴一下测试图看下