BRNSystems/bendzov_image_algorytmus
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

bendzo fixed stuff

Browse Source
This commit is contained in:
Bruno Rybársky 2021-09-04 17:50:13 +02:00
parent ecac7eaa7e
commit a05871e003
2 changed files with 22 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
.idea/vcs.xml Normal file
View File

@ -0,0 +1,6 @@
<?xml version="1.0" encoding="UTF-8"?>
<project version="4">
<component name="VcsDirectoryMappings">
<mapping directory="$PROJECT_DIR$" vcs="Git" />
</component>
</project>

16
minimal_energy.py
View File

@ -3,11 +3,11 @@ from numpy import array as a
import numpy as np import numpy as np
from PIL import Image from PIL import Image
im = Image.open("image.jpg") im = Image.open("bendzo.jpg")
gray = im.convert('LA') gray = im.convert('LA')
asnumpy_gray = np.asarray(gray) asnumpy_gray = np.asarray(gray)
image = np.asarray(im) image = np.asarray(im)
bnsdfcnsd = np.asarray(im)
sobel_0 = [ sobel_0 = [
[-0.125, 0, 0.125], [-0.125, 0, 0.125],
[-0.25, 0, 0.25], [-0.25, 0, 0.25],
@ -67,9 +67,9 @@ def seam_carve_vertical(image, minimal_energy_map):
seam[i + 1] = j + sub_array.index(min(sub_array)) - 1 seam[i + 1] = j + sub_array.index(min(sub_array)) - 1
for i in range(len(seam)): for i in range(len(seam)):
image[i] = image[i].tolist()[:seam[i]] + image[i].tolist()[seam[i] + 1:] image[i] = image[i][:seam[i]] + image[i][seam[i] + 1:]
return image return a([a(row, dtype=np.uint8) for row in image], dtype=np.uint8)
def get_edges_values(input): def get_edges_values(input):
@ -96,6 +96,7 @@ def dostuff(input):
return input return input
for _ in range(100):
tmp = get_minimal_energy_map_vertical(get_edges_values(asnumpy_gray)) tmp = get_minimal_energy_map_vertical(get_edges_values(asnumpy_gray))
tmp = dostuff(tmp) tmp = dostuff(tmp)
@ -104,4 +105,11 @@ image = seam_carve_vertical(image, tmp)
out = Image.fromarray(image) out = Image.fromarray(image)
image = [row.tolist() for row in image]
gray = out.convert('LA')
asnumpy_gray = np.asarray(gray)
asnumpy_gray = a([a([pixel[0] for pixel in row]) for row in asnumpy_gray])
out.show() out.show()