Divide input image color bands by their standard deviation after mean…

… centering. Seems to have a small positive effect on training.
Bleyddyn · Dec 11, 2017 · 94cf17c · 94cf17c
1 parent 803c723
commit 94cf17c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tf/drive_train.py b/tf/drive_train.py
@@ -98,6 +98,9 @@ def loadData( dirs, image_norm=True ):
         images[:,:,:,0] -= np.mean(images[:,:,:,0])
         images[:,:,:,1] -= np.mean(images[:,:,:,1])
         images[:,:,:,2] -= np.mean(images[:,:,:,2])
+        images[:,:,:,0] /= np.std(images[:,:,:,0])
+        images[:,:,:,1] /= np.std(images[:,:,:,1])
+        images[:,:,:,2] /= np.std(images[:,:,:,2])
 
     y = embedActions( actions )
     y = to_categorical( y, num_classes=5 )