Verified Commit f5792788 authored by Max Ehrlich's avatar Max Ehrlich
Browse files

Fix decoding bug

parent 35b472e4
...@@ -25,8 +25,8 @@ def encode(batch, block_size=(8, 8), device=None): ...@@ -25,8 +25,8 @@ def encode(batch, block_size=(8, 8), device=None):
def decode(batch, device=None): def decode(batch, device=None):
block_size = np.sqrt(batch.shape[4]) block_size = int(np.sqrt(batch.shape[4]))
image_size = (batch.shape[2] * block_size, batch.shape[3] * block_size) image_size = (int(batch.shape[2] * block_size), int(batch.shape[3] * block_size))
_, J_i = codec(image_size, (block_size, block_size)) _, J_i = codec(image_size, (block_size, block_size))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment