error about fasterRcnn

[ainilian]

ValueError Traceback (most recent call last)
in ()
5 plt.savefig('image_demo_random.png')
6
----> 7 model.load_weights('weights/faster_rcnn.h5', by_name=True)

~/software/anaconda3/envs/tf2/lib/python3.6/site-packages/tensorflow_core/python/keras/engine/training.py in load_weights(self, filepath, by_name, skip_mismatch)
232 raise ValueError('Load weights is not yet supported with TPUStrategy '
233 'with steps_per_run greater than 1.')
--> 234 return super(Model, self).load_weights(filepath, by_name, skip_mismatch)
235
236 @trackable.no_automatic_dependency_tracking

~/software/anaconda3/envs/tf2/lib/python3.6/site-packages/tensorflow_core/python/keras/engine/network.py in load_weights(self, filepath, by_name, skip_mismatch)
1218 if by_name:
1219 hdf5_format.load_weights_from_hdf5_group_by_name(
-> 1220 f, self.layers, skip_mismatch=skip_mismatch)
1221 else:
1222 hdf5_format.load_weights_from_hdf5_group(f, self.layers)

~/software/anaconda3/envs/tf2/lib/python3.6/site-packages/tensorflow_core/python/keras/saving/hdf5_format.py in load_weights_from_hdf5_group_by_name(f, layers, skip_mismatch)
775 symbolic_weights[i])) +
776 ', but the saved weight has shape ' +
--> 777 str(weight_values[i].shape) + '.')
778
779 else:

ValueError: Layer #4 (named "b_box_head"), weight <tf.Variable 'faster_rcnn_1/b_box_head/rcnn_class_logits/kernel:0' shape=(1024, 21) dtype=float32, numpy=
array([[ 0.04120427, -0.04799104, -0.04488889, ..., -0.05834993,
0.02002645, -0.02086098],
[ 0.0270995 , 0.04724167, 0.072772 , ..., 0.02565087,
0.0062356 , 0.01444595],
[ 0.06293212, 0.07488425, 0.00248578, ..., 0.03330924,
-0.01808247, -0.00931749],
...,
[ 0.02366448, 0.05037834, 0.0206476 , ..., -0.07045348,
0.03336326, -0.01885421],
[ 0.0645275 , 0.02696343, -0.05525204, ..., 0.05430315,
0.02212766, -0.04695521],
[ 0.01637869, -0.00132833, 0.03295995, ..., -0.0239434 ,
0.01851077, 0.04825734]], dtype=float32)> has shape (1024, 21), but the saved weight has shape (1024, 81).