mmocr使用踩坑记录

南怀

已于 2022-04-04 21:44:53 修改

阅读量1.7k

点赞数 1

分类专栏： mmocr 文章标签： python opencv 计算机视觉

于 2021-12-06 17:44:21 首次发布

本文链接：https://blog.csdn.net/qq_38365929/article/details/121163396

版权

mmocr 专栏收录该内容

8 篇文章 1 订阅

订阅专栏

最近在用mmocr修改模型，下面是一些踩坑记录：

1.opencv版本问题

File “/home/mmocr/models/textdet/postprocess/wrapper.py”, line 451, in fcenet_decode
cv2.CHAIN_APPROX_SIMPLE) # opencv4
ValueError: too many values to unpack (expected 2)
出现这个错误就是你的opencv版本要改成4以上的，
pip install opencv-python==4.0.0.21

2.骨干网络

注册表里发现没有SwinTransformer，
1.应该是configs里的backbone里的type写错了，写成这个格式： type=‘mmdet.SwinTransformer’,
因为mmcv里没有SwinTransformer，SwinTransformer只属于mmdet。
2.或者自己创建一个骨干网络，独属于mmocr，
在mmocr/models/textdet/backbones里创建，记得要注册到mmocr的backbones中。

3.AttributeError: ‘NoneType’ object has no attribute ‘shape’

mlt2017数据集进行转换的时候出现了上述问题。原因是数据集中有的照片没有进行转码，只是更改了后缀，所以在模型加载的时候会报错，用PIL工具将所有的gif图片转换成为png图片就可以了
训练集是img_401.gif、img_478.gif、img_1187.gif、img_1188.gif出错
验证集我错的也是这几个序号，？？？

全部转换

import PIL.Image
import os
i=0
path = "/home/sys1/ycs/mmocr/data/mlt2017/imgs/test1/"
savepath = "/home/sys1/ycs/mmocr/data/mlt2017/imgs/test/"
filelist = os.listdir(path)
for file in filelist:
    im = PIL.Image.open(path+filelist[i])
    filename = os.path.splitext(file)[0]
    im.save(savepath+filename+'.png')
    i=i+1
    print(i)

单个转换

import PIL.Image
im = PIL.Image.open('conver/img_401.jpg')
im.save('img_401.jpg')

4.数据集

ICDAR竞赛的数据集在以下网址：
https://rrc.cvc.uab.es/?ch=16&com=downloads
只要注册个账号就可以下载，记得要转换mmocr里的格式。

5.python mmocr/utils/ocr.py

python mmocr/utils/ocr.py 出现
mmocr/core/visualize.py:181: UserWarning: 0 text found in demo/img.jpg
warnings.warn('0 text found in ’ + out_file)
应该是pth文件出错，重新训练得到权重就可