wassname/DenseNet-Keras
1
0
Fork 0
mirror of https://github.com/wassname/DenseNet-Keras.git synced 2026-06-27 17:46:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Updated to Keras (2.0.5) and Tensorflow (1.2.1)

Browse Source
This commit is contained in:
Felix Yu
2017-07-02 19:20:04 +08:00
parent ccbd5cef96
commit 3440f42a89
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
custom_layers.py
+6 -4
View File
@@ -1,6 +1,8 @@
from keras import initializations
from keras.engine import Layer, InputSpec
try:
from keras import initializations
except ImportError:
from keras import initializers as initializations
import keras.backend as K
class Scale(Layer):
@@ -45,8 +47,8 @@ class Scale(Layer):
self.input_spec = [InputSpec(shape=input_shape)]
shape = (int(input_shape[self.axis]),)
self.gamma = self.gamma_init(shape, name='{}_gamma'.format(self.name))
self.beta = self.beta_init(shape, name='{}_beta'.format(self.name))
self.gamma = K.variable(self.gamma_init(shape), name='{}_gamma'.format(self.name))
self.beta = K.variable(self.beta_init(shape), name='{}_beta'.format(self.name))
self.trainable_weights = [self.gamma, self.beta]
if self.initial_weights is not None: