commit #49: add missing import & check GPU usage

Author: BIGBALLON

2_Network_in_Network/Network_in_Network_bn_keras.py

@@ -18,6 +18,7 @@
 weight_decay  = 0.0001
 log_filepath  = './nin_bn'
 
+from keras import backend as K
 if('tensorflow' == K.backend()):
     import tensorflow as tf
     from keras.backend.tensorflow_backend import set_session

2_Network_in_Network/Network_in_Network_keras.py

@@ -18,6 +18,7 @@
 weight_decay  = 0.0001
 log_filepath  = './nin'
 
+from keras import backend as K
 if('tensorflow' == K.backend()):
     import tensorflow as tf
     from keras.backend.tensorflow_backend import set_session

3_Vgg19_Network/Vgg19_keras.py

@@ -28,6 +28,7 @@
 weight_decay = 0.0001
 log_filepath = r'./vgg19_retrain_logs/'
 
+from keras import backend as K
 if('tensorflow' == K.backend()):
     import tensorflow as tf
     from keras.backend.tensorflow_backend import set_session

6_ResNeXt/ResNeXt_keras.py

@@ -27,9 +27,8 @@
 mean = [125.3, 123.0, 113.9]
 std  = [63.0,  62.1,  66.7]
 
-from keras import backend as K
 
-# set GPU memory 
+from keras import backend as K
 if('tensorflow' == K.backend()):
     import tensorflow as tf
     from keras.backend.tensorflow_backend import set_session

7_DenseNet/DenseNet_keras.py

@@ -1,7 +1,6 @@
 import keras
 import math
 import numpy as np
-import keras.backend as K
 from keras.datasets import cifar10
 from keras.preprocessing.image import ImageDataGenerator
 from keras.layers.normalization import BatchNormalization
@@ -27,6 +26,7 @@
 mean = [125.307, 122.95, 113.865]
 std  = [62.9932, 62.0887, 66.7048]
 
+from keras import backend as K
 if('tensorflow' == K.backend()):
     import tensorflow as tf
     from keras.backend.tensorflow_backend import set_session

8_SENet/SENet_Keras.py

@@ -11,7 +11,6 @@
 from keras.models import Model
 from keras import optimizers
 from keras import regularizers
-from keras import backend as K
 
 cardinality        = 4          # 4 or 8 or 16 or 32
 base_width         = 64
@@ -29,6 +28,7 @@
 mean = [125.307, 122.95, 113.865]
 std  = [62.9932, 62.0887, 66.7048]
 
+from keras import backend as K
 if('tensorflow' == K.backend()):
     import tensorflow as tf
     from keras.backend.tensorflow_backend import set_session