整治PPOCRLabel中cv2文件读取问题(更新中)

PPOCRLabel

  • 使用PPOCRLabel对ocr预标注结果进行纠正
  • 由于PaddleOCR代码库十分混乱,路径经常乱掉pip和代码库的代码,经常报错,ocr也是使用paddleocr的pip包;
安装
shell 复制代码
pip install PPOCRLabel==2.1.3
启动
shell 复制代码
PPOCRLabel --lang ch

修改记录

修改1:
  • 报错:AttributeError: 'NoneType' object has no attribute 'shape'
angular2html 复制代码
'NoneType' object has no attribute 'shape'
Traceback (most recent call last):
  File "D:\ProgramData\Anaconda3\envs\paddle\lib\site-packages\PPOCRLabel\libs\autoDialog.py", line 41, in run
    h, w, _ = cv2.imdecode(np.fromfile(Imgpath, dtype=np.uint8), 1).shape
AttributeError: 'NoneType' object has no attribute 'shape'
  • 原因:cv2不支持读取带有中文字符路径文件
python 复制代码
  h, w, _ = cv2.imdecode(np.fromfile(Imgpath, dtype=np.uint8), 1).shape
  • 解决:使用PIL替换cv2, 将上述代码替换为
python 复制代码
# 修改文件:"D:\ProgramData\Anaconda3\envs\paddle\lib\site-packages\PPOCRLabel\libs\autoDialog.py", line 41
# 增加PIL读取图片函数
from PIL import Image
def load_image(image_path: str, return_chw: bool = True, size: tuple = None):
    image = Image.open(image_path).convert("RGB")
    if size is not None:
        image = image.resize(size)  # resize image
    image = np.asarray(image)
    image = image[:, :, ::-1]  # flip color channels from RGB to BGR
    w, h = image.shape[1], image.shape[0]  # update size after resize
    if return_chw:
        image = image.transpose(2, 0, 1)
    return image, (w, h)

# 替换上面代码
try:
    image_data, (w, h) = load_image(Imgpath, return_chw=False)
except Exception as e:
    print(f"load file {Imgpath} fail!")
    continue
修改2
  • 报错:AttributeError: 'NoneType' object has no attribute 'shape'
python 复制代码
Traceback (most recent call last):
  File "D:\ProgramData\Anaconda3\envs\paddle\lib\site-packages\PPOCRLabel\PPOCRLabel.py", line 1889, in saveFile
    self._saveFile(imgidx, mode=mode)
  File "D:\ProgramData\Anaconda3\envs\paddle\lib\site-packages\PPOCRLabel\PPOCRLabel.py", line 1934, in _saveFile
    self.openNextImg()
  File "D:\ProgramData\Anaconda3\envs\paddle\lib\site-packages\PPOCRLabel\PPOCRLabel.py", line 1880, in openNextImg
    self.loadFile(filename)
  File "D:\ProgramData\Anaconda3\envs\paddle\lib\site-packages\PPOCRLabel\PPOCRLabel.py", line 1550, in loadFile
    height, width, depth = cvimg.shape
AttributeError: 'NoneType' object has no attribute 'shape'
  • 原因还是cv2不能读取中文路径文件
  • 解决:
python 复制代码
# 修改: File "D:\ProgramData\Anaconda3\envs\paddle\lib\site-packages\PPOCRLabel\PPOCRLabel.py", line 1550, in loadFile
# cvimg = cv2.imdecode(np.fromfile(unicodeFilePath, dtype=np.uint8), 1)
cvimg, _ = load_image(unicodeFilePath, return_chw=False)
修改3
  • 报错:error: (-215:Assertion failed) _src.total() > 0 in function 'cv::warpPerspective'

    • 报错描述:在对PPOCRLABEL的框进行重新识别是,发生如下报错:

      复制代码
      Can not recognise the detection box in xxxx,png. Please change manually'

      unicodeFilePath is J:\data\mllm-data\xxxxxxxxx\wKh2CWERPJOAY2x-AAE62o598k0620.png
      OpenCV(4.2.0) C:\projects\opencv-python\opencv\modules\imgproc\src\imgwarp.cpp:3143: error: (-215:Assertion failed) _src.total() > 0 in function 'cv::warpPerspective'

    • 原因是我们的ocr预标注数据Label.txt是使用PIL读取图片数据,调用ppocr进行生产的(并不是在PPOCRLabel工具内部生产的), 当我们修改数据框后, PPOCRLabel尝试再次使用cv2进行读取原图,此时由于cv2对路径较为敏感,经常会读取文件失败,才会出现如上情况

  • 解决:

    依旧是修改PPOCRLabel源码, 把cv2读取改为PIL读取,就不惯着cv2的臭毛病...

python 复制代码
# 修改如下代码
 def reRecognition(self):
    #img = cv2.imdecode(np.fromfile(self.filePath,dtype=np.uint8),1)
    img, _ = load_image(self.filePath, return_chw=False)
相关推荐
小磊哥er2 分钟前
【办公自动化】如何使用Python脚本自动化处理音频?
python
深耕AI4 分钟前
PyTorch自定义模型结构详解:从基础到高级实践
人工智能·pytorch·python
Twilight-pending5 分钟前
计算机系统性能、架构设计、调度策略论文分类体系参考
人工智能·云原生·分类·数据挖掘
格林威8 分钟前
工业检测机器视觉为啥非用工业相机?普通相机差在哪?
人工智能·数码相机·yolo·计算机视觉·视觉检测·相机
索迪迈科技13 分钟前
深度解析:从DeepSeek V3.1到K2 Think的“专才”模型架构
人工智能·ai·语言模型
工藤学编程14 分钟前
零基础学AI大模型之从0到1调用大模型API
人工智能
先做个垃圾出来………18 分钟前
Dify开源AI框架介绍
人工智能·开源
带娃的IT创业者22 分钟前
《AI大模型应知应会100篇》第68篇:移动应用中的大模型功能开发 —— 用 React Native 打造你的语音笔记摘要 App
人工智能·笔记·react native
skywalk816331 分钟前
copyparty 是一款使用单个 Python 文件实现的内网文件共享工具,具有跨平台、低资源占用等特点,适合需要本地化文件管理的场景
开发语言·python
Godspeed Zhao31 分钟前
自动驾驶中的传感器技术42——Radar(3)
人工智能·机器学习·自动驾驶