diff --git a/server/visualdl/lib.py b/server/visualdl/lib.py
index 4aae5b5129f1620babe501f1013b525387db77f0..e26c955c31903558b5e96ce3ad3e0fb504989891 100644
--- a/server/visualdl/lib.py
+++ b/server/visualdl/lib.py
@@ -126,7 +126,8 @@ def get_invididual_image(storage, mode, tag, step_index, max_size=80):
             size = max(shape[0], shape[1])
             if size > max_size:
                 scale = max_size * 1. / size
-                im = im.resize(shape[:2])
+                scaled_shape = (shape[0]*scale, shape[1]*scale)
+                im = im.resize(scaled_shape)
             im.save(tempfile)
         tempfile.seek(0, 0)
         return tempfile