search

KichangKim / DeepDanbooru

AI based multi-label girl image classification system, implemented by using TensorFlow.
MIT License
2.6k stars 259 forks source link

wrong params given to layers in squeeze_excitation #55

Closed wranai closed 2 years ago

wranai commented 2 years ago

This here for squeeze_excitation seems to be incorrect: s is just getting overwritten in each line since the layers work on x.


    s = x

    s = tf.keras.layers.GlobalAveragePooling2D()(s)
    s = tf.keras.layers.Dense(output_filters // reduction, activation="relu")(x)
    s = tf.keras.layers.Dense(output_filters, activation="sigmoid")(x)
    x = tf.keras.layers.Multiply()([x, s])
KichangKim commented 2 years ago

Thanks for report. Fixed by 1f6774dd9a7a89d042fb0ae2940e8f12ba845b54