keras-resnet3d icon indicating copy to clipboard operation
keras-resnet3d copied to clipboard

Import Format - Batch Normalization

Open pedr0sorio opened this issue 2 years ago • 1 comments

It seems that there is an issue when importing 'BatchNormalization' from 'keras.layers.normalization' in the script [resnet3d.py]

The import should be done in a different way:

from tensorflow.keras.layers import BatchNormalization

as opposed to

from keras.layers.normalization import BatchNormalization

Maybe the issue is due to some underlying problem.

I am installing this package with the command below in google colab

! pip install git+https://github.com/JihongJu/keras-resnet3d.git and creating a modle like so:

from resnet3d import Resnet3DBuilder
model = Resnet3DBuilder.build_resnet_50((96, 96, 96, 1), 20)

pedr0sorio avatar Apr 28 '22 14:04 pedr0sorio

Hi @pedr0sorio, the package was made in the Keras 1.0 era. Feel free to send a pull request to make it Tensorflow 2.0 compatible.

JihongJu avatar May 01 '22 01:05 JihongJu