added code for gif

1508.06576/image2gif.py

@@ -10,7 +10,7 @@ for img in os.listdir(path):
 
 fnames.sort()
 
-with imageio.get_writer('gifs/'+newNameFolder+".gif", mode='I',duration = 0.9) as writer:
+with imageio.get_writer('gifs/'+newNameFolder+".gif", mode='I',duration = 0.2) as writer:
     for fname in fnames:
         image = imageio.imread(fname)
         writer.append_data(image)

1508.06576/logger.py

@@ -11,7 +11,7 @@ class Logger:
     _formatter = logging.Formatter('%(asctime)s %(levelname)s %(message)s')
     def __new__(cls,*args,**kwargs):
         if not cls._instance:
-            identifier = 'content-reconstruction-mixed'
+            identifier = ''
             if not os.path.isdir("Logs/"):
                 os.mkdir("Logs/")
             logHandler = logging.FileHandler("Logs/style_transfer_"+identifier+".log")
@@ -35,4 +35,4 @@ if __name__ == "__main__":
     ERROR = b.logger()
     INFO.info("TEST")
     ERROR.info("ERROR")
-'''
\ No newline at end of file
+'''

1508.06576/optimizer.py

@@ -48,4 +48,8 @@ class Optimizer:
             #plt.plot(content_img_clone)
             if(e%10 == 0):
                 LOGGER.info(f"Epoch = {e} Total Loss = {total_loss} content Loss = {total_cont_loss} style Loss = {total_style_loss}")
-                save_image(content_img_clone,"styled.png")
\ No newline at end of file
+                identifier = ""
+                name = "styled_images/"+identifier+"/styled_" + str(e) +".png"
+                save_image(content_img_clone,name)
+
+                save_image(content_img_clone,name)
\ No newline at end of file