search

keras-team / keras-applications

Reference implementations of popular deep learning models.
Other
2k stars 913 forks source link

efficientnet SE modual meet backend.image_data_format trouble #155

Open ZJQbelief opened 4 years ago

ZJQbelief commented 4 years ago

https://github.com/keras-team/keras-applications/blob/976050c468ff949bcbd9b9cf64fe1d5c81db3f3a/keras_applications/efficientnet.py#L175

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!