Region Filling and Object Removal by Exemplar-Based Image Inpainting

# coding:utf-8

import cv2 as cv

img = cv.imread('2.png')
print img.shape
x1 = 30
y1 = 70
x2 = 100
y2 = 170
img[x1:x2, y1:y2] = (0, 0, 0)
SIZE = 5

def E(x1, y1, x2, y2):
    s = 0
    b = SIZE // 2

    for x in xrange(b):
        for y in xrange(SIZE):
            var_f = img[x1 + x - b, y1 + y - b] - img[x2 + x - b, y2 + y - b]
            var_s = var_f[0] ** 2
            var_s += var_f[1] ** 2
            var_s += var_f[2] ** 2

            s += var_s
    return s


def find_pixel(px, py):
    var_p = img[SIZE, SIZE]
    min_E = E(px, py, SIZE, SIZE)
    var_x = var_y = SIZE

    for x in xrange(SIZE, img.shape[0] - SIZE):
        for y in xrange(SIZE, img.shape[1] - SIZE):
            if x1 <= x and x <= x2 and y1 <= y and y <= y2:
                continue

            temp_E = E(px, py, x, y)

            if temp_E < min_E:
                min_E = temp_E
                var_p = img[x, y]
                var_x = x
                var_y = y

    print var_x, var_y
    return var_p

for x in xrange(x1, x2 + 1):
    for y in xrange(y1, y2 + 1):
        img[x, y] = find_pixel(x, y)
        print 'y.', y
    print 'x......', x

cv.imwrite('res-size-' + str(SIZE) +'.jpg', img)
cv.waitKey(0)