sofi xd

#Problema 1
def energia(img, i, j):
    deltaXa = deltaX(img, i, j)

    deltaYa = deltaY(img, i, j)

    energia = deltaXa + deltaYa

    return energia

def deltaX(img, i, j):
    deltaX = 0
    for a in range(0, 3):
        if i + 2 > img.shape[0]:
            deltaX += 0
        else:
            deltaX += (img[i + 1, j, a] - img[i - 1, j, a])**2
    return deltaX

def deltaY(img, i, j):
    deltaY = 0
    for a in range(0, 3):
        if j + 2 > img.shape[1]:
            deltaY += 0
        else:
            deltaY += (img[i, j + 1, a] - img[i, j - 1, a])**2
    return deltaY

#problema 2
def borde(energia, a):
    cup2 = np.zeros(cup.shape, int)
    for i in range(0, cup.shape[0]):
        for j in range(0, cup.shape[1]):
            cup2[i,j,] = 255 if energia(cup, i, j) < a else 0
    return cup2
plt.imshow(borde(energia, 1000), cmap='Greys')