Update enet.py

enet.py

@@ -421,8 +421,9 @@ def ENet(inputs,
              slim.arg_scope([slim.batch_norm], fused=True), \
              slim.arg_scope([slim.conv2d, slim.conv2d_transpose], activation_fn=None): 
             #=================INITIAL BLOCK=================
-            for i in xrange(1, max(num_initial_blocks, 1) + 1):
-                net = initial_block(inputs, scope='initial_block_' + str(i))
+            net = initial_block(inputs, scope='initial_block_1')
+            for i in xrange(2, max(num_initial_blocks, 1) + 1):
+                net = initial_block(net, scope='initial_block_' + str(i))
 
             #Save for skip connection later
             if skip_connections:
@@ -512,4 +513,4 @@ def ENet_arg_scope(weight_decay=2e-4,
     with slim.arg_scope([slim.batch_norm],
                         decay=batch_norm_decay,
                         epsilon=batch_norm_epsilon) as scope:
-      return scope
+      return scope