rlx/personalcomputer
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

fixing c > 1 ...

Browse source
This commit is contained in:
rlx 2021-08-10 14:16:44 +00:00
parent 1beadbdc3d
commit 75f55bf55d
1 changed files with 9 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
pc.py
View file

@ -88,7 +88,8 @@ def read_images():
if not os.path.exists(cell_fn): if not os.path.exists(cell_fn):
image = load_image(src_fn) image = load_image(src_fn)
image.resize((cell_w, cell_h), Image.LANCZOS).save(cell_fn) image.resize((cell_w, cell_h), Image.LANCZOS).save(cell_fn)
if sum(data[i]) == 0: #if sum(data[i]) == 0:
if np.all(data[i]==0):
if not image: if not image:
image = load_image(src_fn) image = load_image(src_fn)
data[i] = np.array(image.resize((cx, cy), Image.LANCZOS).getdata()) data[i] = np.array(image.resize((cx, cy), Image.LANCZOS).getdata())
@ -106,6 +107,9 @@ def write_images(data):
src_fns = [f'{src_dn}/{fn}' for fn in os.listdir(src_dn) if fn[0] != '.'] src_fns = [f'{src_dn}/{fn}' for fn in os.listdir(src_dn) if fn[0] != '.']
src_fns = sorted(src_fns) src_fns = sorted(src_fns)
n = len(src_fns) n = len(src_fns)
if c > 1:
s = data.shape
data = data.reshape(s[0], s[1] * s[2])
tree = cKDTree(data if kf == 1 else [data[i] for i in range(0, n, kf)]) tree = cKDTree(data if kf == 1 else [data[i] for i in range(0, n, kf)])
for i, src_fn in enumerate(src_fns): for i, src_fn in enumerate(src_fns):
if i % ts != ti: if i % ts != ti:
@ -121,6 +125,10 @@ def write_images(data):
for x in range(m): for x in range(m):
crop = (x * cx, y * cy, (x + 1) * cx, (y + 1) * cy) crop = (x * cx, y * cy, (x + 1) * cx, (y + 1) * cy)
cell_data = src_image.crop(crop).getdata() cell_data = src_image.crop(crop).getdata()
if c > 1:
cell_data = np.array(cell_data)
s = cell_data.shape
cell_data = cell_data.reshape(s[0] * s[1])
cell_image = get_image(tree.query(cell_data)[1] * kf) cell_image = get_image(tree.query(cell_data)[1] * kf)
dst_image.paste(cell_image, (x * cell_w, y * cell_h)) dst_image.paste(cell_image, (x * cell_w, y * cell_h))
dst_image.save(dst_fn) dst_image.save(dst_fn)