Error running object_detection/builders/model_builder_tf2_test.py file

Prerequisites

Please answer the following questions for yourself before submitting an issue.

• [√] I am using the latest TensorFlow Model Garden release and TensorFlow 2.
• [√] I am reporting the issue to the correct repository. (Model Garden official or research directory)
• [√] I checked to make sure that this issue has not been filed already.

1. The entire URL of the file you are using

https://github.com/tensorflow/models/tree/master/official/...

2. Describe the bug

[ OK ] ModelBuilderTF2Test.test_unknown_faster_rcnn_feature_extractor
[ RUN ] ModelBuilderTF2Test.test_unknown_meta_architecture
INFO:tensorflow:time(main.ModelBuilderTF2Test.test_unknown_meta_architecture): 0.0s
I1123 15:23:23.062027 140676018898688 test_util.py:1973] time(main.ModelBuilderTF2Test.test_unknown_meta_architecture): 0.0s
[ OK ] ModelBuilderTF2Test.test_unknown_meta_architecture
[ RUN ] ModelBuilderTF2Test.test_unknown_ssd_feature_extractor
INFO:tensorflow:time(main.ModelBuilderTF2Test.test_unknown_ssd_feature_extractor): 0.0s
I1123 15:23:23.063109 140676018898688 test_util.py:1973] time(main.ModelBuilderTF2Test.test_unknown_ssd_feature_extractor): 0.0s
[ OK ] ModelBuilderTF2Test.test_unknown_ssd_feature_extractor

ERROR: test_create_center_net_model (main.ModelBuilderTF2Test)
ModelBuilderTF2Test.test_create_center_net_model
Test building a CenterNet model from proto txt.

Traceback (most recent call last):
File "object_detection/builders/model_builder_tf2_test.py", line 224, in test_create_center_net_model
model = model_builder.build(config, is_training=True)
File "/home/pjy/models-master/research/object_detection/builders/model_builder.py", line 1090, in build
add_summaries)
File "/home/pjy/models-master/research/object_detection/builders/model_builder.py", line 976, in _build_center_net_model
center_net_config.feature_extractor)
File "/home/pjy/models-master/research/object_detection/builders/model_builder.py", line 1054, in _build_center_net_feature_extractor
bgr_ordering=feature_extractor_config.bgr_ordering
File "/home/pjy/models-master/research/object_detection/models/center_net_resnet_feature_extractor.py", line 148, in resnet_v2_101
bgr_ordering=bgr_ordering
File "/home/pjy/models-master/research/object_detection/models/center_net_resnet_feature_extractor.py", line 74, in init
resnet_output = tf.keras.layers.BatchNormalization()(resnet_output)
File "/home/pjy/python3.6/lib/python3.6/site-packages/tensorflow/python/keras/engine/base_layer.py", line 926, in call
input_list)
File "/home/pjy/python3.6/lib/python3.6/site-packages/tensorflow/python/keras/engine/base_layer.py", line 1098, in _functional_construction_call
self._maybe_build(inputs)
File "/home/pjy/python3.6/lib/python3.6/site-packages/tensorflow/python/keras/engine/base_layer.py", line 2643, in _maybe_build
self.build(input_shapes) # pylint:disable=not-callable
File "/home/pjy/python3.6/lib/python3.6/site-packages/tensorflow/python/keras/layers/normalization.py", line 350, in build
input_shape)
ValueError: ('Input has undefined axis dimension. Input shape: ', TensorShape([None, 256, None, None]))

Ran 20 tests in 32.850s

FAILED (errors=1, skipped=1)

3. Steps to reproduce

Test run python3 object_detection/builders/model_builder_tf2_test.py error

4. Expected behavior

I install the api according to the steps, but every time I make an error in this test link, the error problem is this one

5. Additional context

Include any logs that would be helpful to diagnose the problem.

6. System information

• OS Platform and Distribution (e.g., Linux Ubuntu 16.04):
• Mobile device name if the issue happens on a mobile device:
• TensorFlow installed from (source or binary):
• TensorFlow version (use command below):2.3.1
• Python version:3.6.5
• Bazel version (if compiling from source):
• GCC/Compiler version (if compiling from source):
• CUDA/cuDNN version:
• GPU model and memory: