Update animation.py

AniKar · Oct 19, 2020 · 5aba1d3 · 5aba1d3
1 parent d725192
commit 5aba1d3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/source/animation.py b/source/animation.py
@@ -15,7 +15,7 @@
 for i in range(len(img_arr)):
     enc_arr.append(encoder.predict(img_arr[i].reshape(1, 784)))
 
-# create animation by linearly interpolating between latent codes of the numbers.
+# create animation by linearly interpolating between latent codes of the images.
 def create_interpolation_animation(enc1, enc2, fname):
     fig = plt.figure()
     ims = []
@@ -36,4 +36,4 @@ def create_interpolation_animation(enc1, enc2, fname):
 create_interpolation_animation(enc_arr[1], enc_arr[7], odir+'animation_1_7.gif')
 create_interpolation_animation(enc_arr[2], enc_arr[5], odir+'animation_2_5.gif')
 create_interpolation_animation(enc_arr[4], enc_arr[6], odir+'animation_4_6.gif')
-create_interpolation_animation(enc_arr[0], enc_arr[6], odir+'animation_0_6.gif')
+create_interpolation_animation(enc_arr[0], enc_arr[6], odir+'animation_0_6.gif')