inference error

[Vvdinosaur]

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?

[rohitpaul23]

In line 208, check whats the value of bn_axis used