wassname/keras-contrib
1
0
Fork 0
mirror of https://github.com/wassname/keras-contrib.git synced 2026-09-11 12:20:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix https://github.com/titu1994/Keras-NASNet/issues/2

Browse Source
This commit is contained in:
Somshubra Majumdar
2017-11-29 08:56:02 -06:00
parent aeebb8c520
commit 3ab207d8a5
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
keras_contrib/applications/nasnet.py
+7 -3
View File
@@ -29,6 +29,8 @@ from keras.layers import GlobalAveragePooling2D
from keras.layers import GlobalMaxPooling2D
from keras.layers import Conv2D
from keras.layers import SeparableConv2D
from keras.layers import ZeroPadding2D
from keras.layers import Cropping2D
from keras.layers import concatenate
from keras.layers import add
from keras.utils.data_utils import get_file
@@ -262,7 +264,7 @@ def NASNetLarge(input_shape=None,
dropout=0.5,
use_auxilary_branch=False,
include_top=True,
weights=None,
weights='imagenet',
input_tensor=None,
pooling=None,
classes=1000):
@@ -333,7 +335,7 @@ def NASNetMobile(input_shape=None,
dropout=0.5,
use_auxilary_branch=False,
include_top=True,
weights=None,
weights='imagenet',
input_tensor=None,
pooling=None,
classes=1000):
@@ -530,7 +532,9 @@ def _adjust_block(p, ip, filters, id=None):
p1 = Conv2D(filters // 2, (1, 1), padding='same', use_bias=False,
name='adjust_conv_1_%s' % id, kernel_initializer='he_normal')(p1)
p2 = AveragePooling2D((1, 1), strides=(2, 2), padding='valid', name='adjust_avg_pool_2_%s' % id)(p)
p2 = ZeroPadding2D(padding=((0, 1), (0, 1)))(p)
p2 = Cropping2D(cropping=((1, 0), (1, 0)))(p2)
p2 = AveragePooling2D((1, 1), strides=(2, 2), padding='valid', name='adjust_avg_pool_2_%s' % id)(p2)
p2 = Conv2D(filters // 2, (1, 1), padding='same', use_bias=False,
name='adjust_conv_2_%s' % id, kernel_initializer='he_normal')(p2)