Begin frame augmentation

Author: pratikgujjar

code/autoencoder_model/scripts/config_oc.py

@@ -81,7 +81,7 @@
 LOSS_WEIGHTS = [1, 1]
 A_TRAIN_RATIO = 1
 C_TRAIN_RATIO = 1
-RAM_DECIMATE = False
+RAM_DECIMATE = True
 RETRAIN_CLASSIFIER = False
 CLASS_TARGET_INDEX = 8
 
@@ -97,8 +97,8 @@
 
 # simple_ped_set = ['crossing', 'stopped', 'looking', 'clear path', 'unknown']
 
-# simple_ped_set = ['standing 0', 'approaching 1', 'looking 2', 'crossing 3', 'stopped 4', 'clear path 5', 'unknown 6', 'no ped 7']
-simple_ped_set = ['standing', 'approaching', 'looking', 'crossing', 'stopped', 'clear path', 'unknown', 'no ped']
+# simple_ped_set = ['standing 0', 'approaching 1', 'crossing 2', 'stopped 3', 'clear path 4', 'unknown 5', 'no ped 6']
+simple_ped_set = ['standing', 'approaching', 'crossing', 'stopped', 'clear path', 'unknown', 'no ped']
 
 
 
@@ -125,11 +125,11 @@
 print ("Config file: " + str(__name__))
 
 BATCH_SIZE = 25
-NB_EPOCHS_CLASS = 100
+NB_EPOCHS_CLASS = 15
 
 # OPTIM_C = Adam(lr=0.0000002, beta_1=0.5)
-OPTIM_C = SGD(lr=0.0001, momentum=0.9, nesterov=True)
-# OPTIM_C = RMSprop(lr=0.0001, rho=0.9)
+# OPTIM_C = SGD(lr=0.0001, momentum=0.9, nesterov=True)
+OPTIM_C = RMSprop(lr=0.0001, rho=0.9)
 
 # lr_schedule = [10, 20, 30]  # epoch_step
 
@@ -143,12 +143,12 @@
 #     return 0.000000001
 
 
-lr_schedule = [5, 10, 15]  # epoch_step
+lr_schedule = [4, 7, 10]  # epoch_step
 def schedule(epoch_idx):
     if (epoch_idx + 1) < lr_schedule[0]:
-        return 0.00001
+        return 0.0001
     elif (epoch_idx + 1) < lr_schedule[1]:
-        return 0.000001  # lr_decay_ratio = 10
+        return 0.00001  # lr_decay_ratio = 10
     elif (epoch_idx + 1) < lr_schedule[2]:
-        return 0.0000001
-    return 0.0000001
+        return 0.000001
+    return 0.000001

code/autoencoder_model/scripts/config_r16.py

@@ -69,7 +69,7 @@
 IMG_SIZE = (128, 208, 3)
 ATTN_COEFF = 0
 KL_COEFF = 0
-RAM_DECIMATE = True
+RAM_DECIMATE = False
 REVERSE = True
 
 # -------------------------------------------------
@@ -78,7 +78,7 @@
 print ("Config file: " + str(__name__))
 
 BATCH_SIZE = 10
-NB_EPOCHS_AUTOENCODER = 40
+NB_EPOCHS_AUTOENCODER = 30
 
 OPTIM_A = Adam(lr=0.0001, beta_1=0.5)
 # OPTIM_A = SGD(lr=0.000001, momentum=0.5, nesterov=True)

code/autoencoder_model/scripts/gifimages.py

@@ -11,7 +11,7 @@ def create_gif(filenames, duration, vid_num):
     imageio.mimsave(os.path.join(GIF_DIR, video_name), images, duration=duration)
 
 
-def strip(image, img_height, img_width, vid_len, vid_num=1):
+def strip(image, img_height, img_width, vid_len, vid_num=1, rev=False):
     n_rows = image.shape[0]
     n_cols = image.shape[1]
     n_horizontal_imgs = n_cols/img_width
@@ -34,8 +34,9 @@ def strip(image, img_height, img_width, vid_len, vid_num=1):
             filename = "vid_" + str(vid_num) + "_frame_" + str(frame_num) + ".png"
             cv2.imwrite(os.path.join(GIF_IMG_DIR, filename), img)
             filenames.append(os.path.join(GIF_IMG_DIR, filename))
-            if frame_num == (int(vid_len/2)):
-                filenames = list(reversed(filenames))
+            if rev:
+                if frame_num == (int(vid_len/2)):
+                    filenames = list(reversed(filenames))
             if frame_num == vid_len:
 
                 create_gif(filenames=filenames, duration=duration, vid_num=vid_num)
@@ -60,6 +61,7 @@ def get_args():
     parser.add_argument("--img_height", type=int, default=128)
     parser.add_argument("--img_width", type=int, default=128)
     parser.add_argument("--vid_len", type=int, default=10)
+    parser.add_argument("--rev", type=bool, default=False)
     args = parser.parse_args()
     return args
 
@@ -87,4 +89,4 @@ def get_args():
         except IOError as e:
             print (e)
 
-        strip(image=im, img_height=args.img_height, img_width=args.img_width, vid_len=args.vid_len)
+        strip(image=im, img_height=args.img_height, img_width=args.img_width, vid_len=args.vid_len, rev=args.rev)