Dilated convolutions in encoder

Author: pratikgujjar

code/autoencoder_model/scripts/argparse

@@ -62,7 +62,7 @@
 
 PRINT_MODEL_SUMMARY = True
 SAVE_MODEL = True
-PLOT_MODEL = False
+PLOT_MODEL = True
 SAVE_GENERATED_IMAGES = True
 SHUFFLE = True
 VIDEO_LENGTH = 32
@@ -77,7 +77,7 @@
 print ("Loading network/training configuration.")
 print ("Config file: " + str(__name__))
 
-BATCH_SIZE = 10
+BATCH_SIZE = 9
 NB_EPOCHS_AUTOENCODER = 30
 
 OPTIM_A = Adam(lr=0.0001, beta_1=0.5)

code/autoencoder_model/scripts/config_r16.py

@@ -46,10 +46,10 @@ def strip(image, img_height, img_width, vid_len, vid_num=1, rev=False):
             frame_num = frame_num + 1
 
 
-def gifmax(folder):
+def gifmax(folder, rev=False):
     for i in range(880):
-        im_pred = cv2.imread(folder+'/truth/' + str(i) + '_truth.png', cv2.IMREAD_COLOR)
-        strip(im_pred, 128, 32, i)
+        im_pred = cv2.imread(folder+'/pred/' + str(i) + '_pred.png', cv2.IMREAD_COLOR)
+        strip(im_pred, 128, 208, 32, i, rev=rev)
 
 
 def get_args():
@@ -79,7 +79,7 @@ def get_args():
     args = get_args()
     if args.mode == 'gifmax':
         print ('going into gifmax')
-        gifmax(args.folder)
+        gifmax(args.folder, args.rev)
     else:
         try:
             im = cv2.imread(args.file, cv2.IMREAD_COLOR)