add densenet

zz44l43 · Dec 13, 2017 · f03b007 · f03b007
1 parent ab806a3
commit f03b007
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/nets/nets_factory.py b/nets/nets_factory.py
@@ -30,6 +30,7 @@
 from nets import resnet_v1
 from nets import resnet_v2
 from nets import vgg
+from nets import densenet
 from nets.nasnet import nasnet
 
 slim = tf.contrib.slim
@@ -61,6 +62,7 @@
                 'nasnet_cifar': nasnet.build_nasnet_cifar,
                 'nasnet_mobile': nasnet.build_nasnet_mobile,
                 'nasnet_large': nasnet.build_nasnet_large,
+                'densenet': densenet.densenet,
                }
 
 arg_scopes_map = {'alexnet_v2': alexnet.alexnet_v2_arg_scope,
@@ -91,6 +93,7 @@
                   'nasnet_cifar': nasnet.nasnet_cifar_arg_scope,
                   'nasnet_mobile': nasnet.nasnet_mobile_arg_scope,
                   'nasnet_large': nasnet.nasnet_large_arg_scope,
+                  'densenet': densenet.densenet_arg_scope,
                  }
 
 

diff --git a/preprocessing/preprocessing_factory.py b/preprocessing/preprocessing_factory.py
@@ -68,6 +68,7 @@ def get_preprocessing(name, is_training=False):
       'vgg_a': vgg_preprocessing,
       'vgg_16': vgg_preprocessing,
       'vgg_19': vgg_preprocessing,
+      'densenet': inception_preprocessing,
   }
 
   if name not in preprocessing_fn_map: