-
Notifications
You must be signed in to change notification settings - Fork 42
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
inference error #10
Comments
In line 208, check whats the value of bn_axis used |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
when i run "python3 inference_15parts.py --scale=1" , error happens like:
Traceback (most recent call last):
File "inference_15parts.py", line 200, in
model = get_testing_model_resnet101()
File "D:\body\body\model_simulated_RGB101.py", line 456, in get_testing_model_resnet101
C1, C2, C3, C4, C5 = ResNet101_graph(img_input, None)
File "D:\body\body\model_simulated_RGB101.py", line 250, in ResNet101_graph
x = conv_block(x, 3, [64, 64, 256], stage=2, block='a', strides=(1, 1))
File "D:\body\body\model_simulated_RGB101.py", line 208, in conv_block
x = BatchNormalization(epsilon=eps, axis=bn_axis,name=bn_name_base + '2a')(x)
File "D:\Python3.6\lib\site-packages\keras\engine\base_layer.py", line 463, in call
self.build(unpack_singleton(input_shapes))
File "D:\Python3.6\lib\site-packages\keras\layers\normalization.py", line 99, in build
str(input_shape) + '.')
ValueError: Axis 1 of input tensor should have a defined dimension but the layer received an input with shape (None, None, None, 64).
Has anyone been in the same situation like this?
The text was updated successfully, but these errors were encountered: