具有要由OCR识别的文本的Deblur图像

时间:2018-02-07 22:09:22

标签: c++ opencv image-processing ocr

我有一个模糊的图像:
1 这是名片的一部分,它是相机拍摄的一个框架,没有适当的焦点。

清除图片如下所示 2 我正在寻找可以给我更好质量的图像的方法,这样图像可以被OCR识别,但也应该非常快。图像没有太多模糊(我想是这样),但对OCR来说并不好。我试过了:

  • 不同种类的HPF,
  • 拉普拉斯,
  • Canny探测器,
  • 形态操作的组合(开放,关闭)。

我也尝试过:

  • 使用Wiener过滤器解卷积,
  • deconvolution和Lucy-Richardson方法。

但要找到合适的PSF(点扩散功能)并不容易。这些方法被认为是有效的,但不够快。我也尝试过FFT,然后使用高斯模板进行IFFT,但结果并不令人满意。我正在寻找一种使用文本去除图像去模糊的一般方法,而不仅仅是这个图像。有人可以帮我解决这个问题吗?我会对任何建议表示感谢。我正在使用OpenCV 3(C ++,有时是Python)。

2 个答案:

答案 0 :(得分:12)

您知道Blind deconvolution吗?

盲卷积是恢复天文图像的一项众所周知的技术。这对于难以找到PSF的应用程序特别有用。

Here是此技术的一种C ++实现。这个paper也与您要寻找的东西非常相关。这是他们算法的示例输出:

enter image description here

答案 1 :(得分:7)

我最近也遇到了这个问题,并提出了一个类似的question,其中包含更多详细信息和最新方法。到目前为止,这似乎尚未解决。最近有一些研究工作试图通过深度学习解决此类问题。不幸的是,没有一件作品达到我们的期望。但是,我正在共享信息,以防对任何人有帮助。

1。场景文本图像在野外的超分辨率

对于我们来说,这可能是我们的最后选择;比较而言,表现还不错。最近的研究工作(TSRN)主要针对此类情况。它的主要直观之处是引入超分辨率(SR)技术作为预处理。这个implementation看起来是最有前途的。这是他们取得的成就的例证,可以改善模糊效果,从而获得清晰的图像。

2。神经增强

从他们的repo演示中可以看出,它也可能具有改善模糊文本的潜力。但是,作者可能大约四年都没有维护该回购协议。

3。 GAN进行盲运动去模糊

有吸引力的部分是其中的盲运动去模糊机制,名为DeblurGAN。看起来很有希望。

enter image description here

4。通过内核估计和噪声注入实现现实世界的超分辨率

关于their work的一个有趣的事实是,与其他文学作品不同,他们首先通过估计各种模糊内核,为现实世界的图像设计了一个新颖的降级框架。真实的噪声分布。基于此,他们获取与真实世界图像共享同一域的 LR 图像。然后,他们提出了一个现实世界的超分辨率模型,旨在更好的感知。从他们的文章中:

enter image description here enter image description here

但是,据我观察,我无法获得预期的结果。我提出了一个issue on github,直到现在都没有得到任何回应。


用于直接文本去模糊的卷积神经网络

@Ali共享的paper看起来非常有趣,并且效果非常好。很高兴他们共享了训练有素的模型的预训练权重,并且还共享了python脚本以便于使用。但是,他们已经尝试使用 Caffe 库。我希望转换为 PyTorch 以更好地控制。以下是提供的带有 Caffe 导入的python脚本。请注意,由于缺乏Caffe知识,直到现在我还不能完全移植它。如果您知道它,请更正我。

from __future__ import print_function
import numpy as np
import os, sys, argparse, glob, time, cv2, Queue, caffe

# Some Helper Functins 
def getCutout(image, x1, y1, x2, y2, border):
    assert(x1 >= 0 and y1 >= 0)
    assert(x2 > x1 and y2 >y1)
    assert(border >= 0)
    return cv2.getRectSubPix(image, (y2-y1 + 2*border, x2-x1 + 2*border), (((y2-1)+y1) / 2.0, ((x2-1)+x1) / 2.0))

def fillRndData(data, net):
    inputLayer = 'data'
    randomChannels = net.blobs[inputLayer].data.shape[1]
    rndData = np.random.randn(data.shape[0], randomChannels, data.shape[2], data.shape[3]).astype(np.float32) * 0.2
    rndData[:,0:1,:,:] = data
    net.blobs[inputLayer].data[...] = rndData[:,0:1,:,:]

def mkdirp(directory):
    if not os.path.isdir(directory):
        os.makedirs(directory)

主要功能从这里开始

def main(argv):
    pycaffe_dir = os.path.dirname(__file__)

    parser = argparse.ArgumentParser()
    # Optional arguments.
    parser.add_argument(
        "--model_def",
        help="Model definition file.",
        required=True
    )
    parser.add_argument(
        "--pretrained_model",
        help="Trained model weights file.",
        required=True
    )
    parser.add_argument(
        "--out_scale",
        help="Scale of the output image.",
        default=1.0,
        type=float
    )
    parser.add_argument(
        "--output_path",
        help="Output path.",
        default=''
    )
    parser.add_argument(
        "--tile_resolution",
        help="Resolution of processing tile.",
        required=True,
        type=int
    )
    parser.add_argument(
        "--suffix",
        help="Suffix of the output file.",
        default="-deblur",
    )
    parser.add_argument(
        "--gpu",
        action='store_true',
        help="Switch for gpu computation."
    )
    parser.add_argument(
        "--grey_mean",
        action='store_true',
        help="Use grey mean RGB=127. Default is the VGG mean."
    )
    parser.add_argument(
        "--use_mean",
        action='store_true',
        help="Use mean."
    )
    parser.add_argument(
        "--adversarial",
        action='store_true',
        help="Use mean."
    )
    args = parser.parse_args()

    mkdirp(args.output_path)

    if hasattr(caffe, 'set_mode_gpu'):
        if args.gpu:
            print('GPU mode', file=sys.stderr)
            caffe.set_mode_gpu()
        net = caffe.Net(args.model_def, args.pretrained_model, caffe.TEST)
    else:
        if args.gpu:
            print('GPU mode', file=sys.stderr)
        net = caffe.Net(args.model_def, args.pretrained_model, gpu=args.gpu)


    inputs = [line.strip() for line in sys.stdin]

    print("Classifying %d inputs." % len(inputs), file=sys.stderr)


    inputBlob = net.blobs.keys()[0] # [innat]: input shape 
    outputBlob = net.blobs.keys()[-1]

    print( inputBlob, outputBlob)
    channelCount = net.blobs[inputBlob].data.shape[1]
    net.blobs[inputBlob].reshape(1, channelCount, args.tile_resolution, args.tile_resolution)
    net.reshape()

    if channelCount == 1 or channelCount > 3:
        color = 0
    else:
        color = 1

    outResolution = net.blobs[outputBlob].data.shape[2]
    inResolution = int(outResolution / args.out_scale)
    boundary = (net.blobs[inputBlob].data.shape[2] - inResolution) / 2

    for fileName in inputs:
        img = cv2.imread(fileName, flags=color).astype(np.float32)
        original = np.copy(img)
        img = img.reshape(img.shape[0], img.shape[1], -1)
        if args.use_mean:
            if args.grey_mean or channelCount == 1:
                img -= 127
            else:
                img[:,:,0] -= 103.939
                img[:,:,1] -= 116.779
                img[:,:,2] -= 123.68
        img *= 0.004

        outShape = [int(img.shape[0] * args.out_scale) ,
                    int(img.shape[1] * args.out_scale) ,
                    net.blobs[outputBlob].channels]
        imgOut = np.zeros(outShape)

        imageStartTime = time.time()
        for x, xOut in zip(range(0, img.shape[0], inResolution), range(0, imgOut.shape[0], outResolution)):
            for y, yOut in zip(range(0, img.shape[1], inResolution), range(0, imgOut.shape[1], outResolution)):

                start = time.time()

                region = getCutout(img, x, y, x+inResolution, y+inResolution, boundary)
                region = region.reshape(region.shape[0], region.shape[1], -1)
                data = region.transpose([2, 0, 1]).reshape(1, -1, region.shape[0], region.shape[1])

                if args.adversarial:
                    fillRndData(data, net)
                    out = net.forward()
                else:
                    out = net.forward_all(data=data)

                out = out[outputBlob].reshape(out[outputBlob].shape[1], out[outputBlob].shape[2], out[outputBlob].shape[3]).transpose(1, 2, 0)

                if imgOut.shape[2] == 3 or imgOut.shape[2] == 1:
                    out /= 0.004
                    if args.use_mean:
                        if args.grey_mean:
                            out += 127
                        else:
                            out[:,:,0] += 103.939
                            out[:,:,1] += 116.779
                            out[:,:,2] += 123.68

                if out.shape[0] != outResolution:
                    print("Warning: size of net output is %d px and it is expected to be %d px" % (out.shape[0], outResolution))
                if out.shape[0] < outResolution:
                    print("Error: size of net output is %d px and it is expected to be %d px" % (out.shape[0], outResolution))
                    exit()

                xRange = min((outResolution, imgOut.shape[0] - xOut))
                yRange = min((outResolution, imgOut.shape[1] - yOut))

                imgOut[xOut:xOut+xRange, yOut:yOut+yRange, :] = out[0:xRange, 0:yRange, :]
                imgOut[xOut:xOut+xRange, yOut:yOut+yRange, :] = out[0:xRange, 0:yRange, :]

                print(".", end="", file=sys.stderr)
                sys.stdout.flush()


        print(imgOut.min(), imgOut.max())
        print("IMAGE DONE %s" % (time.time() - imageStartTime))
        basename = os.path.basename(fileName)
        name = os.path.join(args.output_path, basename + args.suffix)
        print(name, imgOut.shape)
        cv2.imwrite( name, imgOut)

if __name__ == '__main__':
    main(sys.argv)

运行程序:

cat fileListToProcess.txt | python processWholeImage.py --model_def ./BMVC_nets/S14_19_200.deploy --pretrained_model ./BMVC_nets/S14_19_FQ_178000.model --output_path ./out/ --tile_resolution 300 --suffix _out.png --gpu --use_mean

重量文件以及上述脚本可以从here (BMVC_net)下载。但是,您可能要转换caffe2pytorch。为此,这是基本的起点:

下一步

# BMVC_net, you need to download it from authors website, link above
model = caffemodel2pytorch.Net(
    prototxt = './BMVC_net/S14_19_200.deploy', 
    weights = './BMVC_net/S14_19_FQ_178000.model',
    caffe_proto = 'https://raw.githubusercontent.com/BVLC/caffe/master/src/caffe/proto/caffe.proto'
)

model.cuda()
model.eval()
torch.set_grad_enabled(False)

运行演示张量,

# make sure to have right procedure of image normalization and channel reordering
image = torch.Tensor(8, 3, 98, 98).cuda()

# outputs dict of PyTorch Variables
# in this example the dict contains the only key "prob"
#output_dict = model(data = image)

# you can remove unneeded layers:
#del model.prob
#del model.fc8

# a single input variable is interpreted as an input blob named "data"
# in this example the dict contains the only key "fc7"
output_dict = model(image)
# print(output_dict)
print(output_dict.keys())

请注意,需要考虑一些基本事项;网络期望DPI 120-150处的文本,合理的方向以及合理的黑白水平。网络期望从输入中减去[103.9、116.8、123.7]。输入应进一步乘以0.004。