提交 27a7df13 编写于 作者: S superjom

fix scalar

上级 f45ff7b9
...@@ -126,7 +126,7 @@ def get_invididual_image(storage, mode, tag, step_index, max_size=80): ...@@ -126,7 +126,7 @@ def get_invididual_image(storage, mode, tag, step_index, max_size=80):
size = max(shape[0], shape[1]) size = max(shape[0], shape[1])
if size > max_size: if size > max_size:
scale = max_size * 1. / size scale = max_size * 1. / size
scaled_shape = (shape[0]*scale, shape[1]*scale) scaled_shape = (int(shape[0]*scale), int(shape[1]*scale))
im = im.resize(scaled_shape) im = im.resize(scaled_shape)
im.save(tempfile) im.save(tempfile)
tempfile.seek(0, 0) tempfile.seek(0, 0)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册