Skip to content

Commit

Permalink
Minor cleaning
Browse files Browse the repository at this point in the history
  • Loading branch information
Manojkumarmuru committed May 12, 2024
1 parent 600f311 commit b086005
Showing 1 changed file with 5 additions and 6 deletions.
11 changes: 5 additions & 6 deletions paz/models/classification/xception.py
Original file line number Diff line number Diff line change
Expand Up @@ -88,9 +88,11 @@ def build_minixception(input_shape, num_classes, l2_reg=0.01):
"""Function for instantiating an Mini-Xception model.
# Arguments
input_shape: List corresponding to the input shape of the model.
input_shape: List corresponding to the input shape
of the model.
num_classes: Integer.
l2_reg. Float. L2 regularization used in the convolutional kernels.
l2_reg. Float. L2 regularization used
in the convolutional kernels.
# Returns
Tensorflow-Keras model.
Expand Down Expand Up @@ -178,12 +180,9 @@ def build_minixception(input_shape, num_classes, l2_reg=0.01):
use_bias=False)(x)
x = BatchNormalization()(x)

# x = MaxPooling2D((3, 3), strides=(1, 1), padding='same')(x)
x = layers.add([x, residual])

x = Conv2D(num_classes, (3, 3),
# kernel_regularizer=regularization,
padding='same')(x)
x = Conv2D(num_classes, (3, 3), padding='same')(x)
x = GlobalAveragePooling2D()(x)
output = Activation('softmax', name='predictions')(x)

Expand Down

0 comments on commit b086005

Please sign in to comment.