Merge pull request #144 from titu1994/master

Refactor DenseNets
2026-06-27 16:10:11 +08:00 · 2017-09-19 08:40:18 -05:00
parent c4d1a61746 ff47da56fb
commit 97c31364cc
2 changed files with 686 additions and 367 deletions
@@ -33,8 +33,11 @@ nb_filter = 16
 dropout_rate = 0.0  # 0.0 for data augmentation

 # Create the model (without loading weights)
-model = DenseNet(depth, nb_dense_block, growth_rate, nb_filter, dropout_rate=dropout_rate,
-                 input_shape=img_dim, weights=None)
+model = DenseNet(depth=depth, nb_dense_block=nb_dense_block,
+                 growth_rate=growth_rate, nb_filter=nb_filter,
+                 dropout_rate=dropout_rate,
+                 input_shape=img_dim,
+                 weights=None)
 print('Model created')

 model.summary()