efficientnet SE modual meet backend.image_data_format trouble

[ZJQbelief]

keras-applications/keras_applications/efficientnet.py

Line 175 in 976050c

se = layers.Reshape((1, 1, filters), name=name + 'se_reshape')(se)

This code will meet trouble when backend.image_data_format() == "channels_first". It should be change as below:

if backend.image_data_format() == 'channels_last':
       se = layers.Reshape((1, 1, filters), name=name + 'se_reshape')(se)
else:
       se = layers.Reshape((filters, 1, 1), name=name + 'se_reshape')(se)

Thank for your working!