Skip to content

Fastdeploy_python初始化cls_model为None之后,推理图片还是会先检测后进行方向旋转然后给识别模型进行识别 #2616

Open
@askxiaozhang

Description

@askxiaozhang

环境

  • 【FastDeploy版本】: fastdeploy_gpu_python-1.0.7-cp39-cp39
  • 【编译命令】通过pip安装的
  • 【系统平台】: Linux x64(Ubuntu 22.04)
  • 【硬件】:Nvidia GPU 4060
  • 【编译语言】:Python(3.9)

ocr初始化模型我是通过该方式初始化的

···
self.ocr = fd.vision.ocr.PPOCRv4(
det_model=det_model, cls_model=None, rec_model=rec_model)
···
然后推理该图片

Image
�推理代码:self.ocr.predict(image)

得到结果
det boxes: [[154,33],[189,33],[189,53],[154,53]]rec text: ALT rec score:0.999999
det boxes: [[347,32],[409,32],[409,53],[347,53]]rec text: 0-40U/L rec score:0.992021
det boxes: [[253,35],[263,35],[263,52],[253,52]]rec text: 三 rec score:0.660740
det boxes: [[14,34],[140,34],[140,51],[14,51]]rec text: 丙氨酸氨基转移酶 rec score:0.999998

然后将[[253,35],[263,35],[263,52],[253,52]]手动裁剪出来

Image
可以看到1这个图片识别效果不好,应该是旋转了。
然后我自己将裁剪的图片进行单独的rec识别
结果如下:
['1', 0.9998661279678345]

该如何解决这个自动旋转的问题? cls_model为None也会自动旋转,导致识别不对

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions